Update app.py

app.py

@@ -184,10 +184,14 @@ async def generate_text(request: Request):
     print("POST Request:")
 
     # Check length of input, if it is greater than 10 tokens, the text is sent off to a summarizer to generate:
-
-    generated_text = generative.generate_account_text(
-        prompt=data["text"], model_dir=os.path.join(models_path, data["account"])
-    )
+    try:
+        generated_text = generative.generate_account_text(
+            prompt=data["text"], model_dir=os.path.join(models_path, data["account"])
+        )
+        logging.info("INFO: Successfully generate text from model.")
+    except Exception as e:
+        logging.error(f"Error generating text: {e}")
+        return {"error": "Error generating text"}
     # return one example
     generated_text = generated_text[0]["generated_text"]