Update main.py

main.py

@@ -34,5 +34,5 @@ async def generate_text(request: Request):
         generated_text = llm(prompt)
         print(generated_text)
         return {'generated_text': generated_text}
-    else:
+    except:
         return {"generated_text": "Error!"}