Update app.py

app.py

@@ -15,12 +15,14 @@ import spaces
 
 # Friendli AI Endpoints parameter
 friendli_token = os.getenv("FRIENDLI_TOKEN", "your_friendli_token")
-
+gemini_token = os.getenv("GEMINI_TOKEN", "your_gemini_token")
 
 # OpenAI client for Friendli
 client = OpenAI(
-    base_url="https://api.friendli.ai/dedicated/v1",
-    api_key=friendli_token
+    # base_url="https://api.friendli.ai/dedicated/v1",
+    # api_key=friendli_token
+    base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
+    api_key=gemini_token
 )
 
 MAX_NUM_IMAGES = int(os.getenv("MAX_NUM_IMAGES", "5"))
@@ -228,9 +230,9 @@ def run(message: dict, history: list[dict]) -> Iterator[str]:
     messages.extend(process_history(history))
     messages.extend(process_new_user_message(message))
 
-    # Generate with Friendli API
+    # Generate with Gemini API
     completion = client.chat.completions.create(
-        model="jlsjb3b0fy8m",  # Use appropriate model
+        model="gemini-2.5-flash-preview-04-17",  # Use appropriate model
         messages=messages,
         max_tokens=2048,
         temperature=0.2,