Update main.py

main.py

@@ -30,6 +30,9 @@ async def generate_text(request: Request):
     prompt = raw_data.decode('utf-8')
     if not prompt:
         return {'error': 'Prompt is required'}, 400
-
-    generated_text = llm(prompt)
-    return {'generated_text': generated_text}
+    try:
+        generated_text = llm(prompt)
+        print(generated_text)
+        return {'generated_text': generated_text}
+    else:
+        return {"generated_text": "Error!"}