Update main.py

main.py

@@ -78,15 +78,10 @@ def handle_message(event):
                             "parts": [message],
                             })
             # Check if response is valid
-            
-            # response = model.generate_content(messages)
-            # out = response.parts[0].text
-            # if not response.parts or not response.parts[0].text: 
-            # out = response.parts[0].text
-
-            out = client.predict(messages, api_name="/predict").parts[0].text
-            if not out:
-                out = "Gemini: [No response or the response does not contain text]"
+            response = model.generate_content(messages)
+            out = response.parts[0].text
+            if not response.parts or not response.parts[0].text: 
+                out = response.parts[0].text
         except:
             out = "Gemini error!"