Update app.py

app.py

@@ -4,6 +4,7 @@ import json
 import re
 import gradio as gr
 import google.generativeai as genai
+import random
 
 print("Google Generative AI SDK version:", genai.__version__)
 
@@ -52,7 +53,7 @@ def process_image(image, prompt):
         if not api_key:
             return "ERROR: Missing GOOGLE_API_KEY.", None
         genai.configure(api_key=api_key)
-        model_name = INTERNAL_MODEL_MAP.get(model_choice, "gemini-2.5-flash")
+        model_name = "gemini-2.5-flash"
         model = genai.GenerativeModel(model_name=model_name,
                                       generation_config={"temperature": float(0.2), "top_p": float(0.95)})
 
@@ -79,12 +80,7 @@ def process_image(image, prompt):
         #uploaded_file = client.files.upload(file=temp_file)
         uploaded_file = genai.upload_file(path=temp_file, mime_type="image/jpeg")
         print("Uploaded:", uploaded_file.name)
-
-        # === CREATE MODEL USING genai.GenerativeModel ====
-        model = genai.GenerativeModel(
-            "gemini-2.0-flash",
-            client=client
-        )
+        
 
         # === GENERATE CONTENT ====
         response = model.generate_content([prompt, uploaded_file])