Update app.py

app.py

@@ -207,7 +207,7 @@ if prompt:
             for i in range(1):
                 print(f"Attempt {i+1}")
                 if model_name == "gemini-pro":
-                    llm = GoogleGenerativeAI(model=model, google_api_key=os.getenv("GEMINI_TOKEN"), temperature=0)
+                    llm = GoogleGenerativeAI(model=models[model_name], google_api_key=os.getenv("GEMINI_TOKEN"), temperature=0)
                 else:
                     llm = ChatGroq(model=models[model_name], api_key=os.getenv("GROQ_API"), temperature=0)