Update main.py

main.py

@@ -28,7 +28,7 @@ app.add_middleware(
 @app.get("/")
 def root():
     def get_gemini_response(input):
-        response = model.generate_content(input)
+        response = model.generate_content(input, generation_config=generation_config)
         return response.text
     # Define the Gradio interface
     iface = gr.Interface(