Update app.py
Browse files
app.py
CHANGED
|
@@ -104,10 +104,10 @@ def get_llm_response(user_text):
|
|
| 104 |
"""Sends text to Gemini -> Gets Text Response"""
|
| 105 |
try:
|
| 106 |
start_time = time.time()
|
| 107 |
-
|
| 108 |
print(f"[Timing] LLM took: {time.time() - start_time:.2f}s")
|
| 109 |
-
|
| 110 |
-
|
| 111 |
except Exception as e:
|
| 112 |
return f"LLM Error: {e}"
|
| 113 |
|
|
|
|
| 104 |
"""Sends text to Gemini -> Gets Text Response"""
|
| 105 |
try:
|
| 106 |
start_time = time.time()
|
| 107 |
+
response = chat_session.send_message(user_text, safety_settings=safety_settings)
|
| 108 |
print(f"[Timing] LLM took: {time.time() - start_time:.2f}s")
|
| 109 |
+
return response.text
|
| 110 |
+
|
| 111 |
except Exception as e:
|
| 112 |
return f"LLM Error: {e}"
|
| 113 |
|