Spaces:
Runtime error
Runtime error
aus10powell
commited on
Commit
•
5d83b09
1
Parent(s):
acf96b5
Update app.py
Browse files
app.py
CHANGED
@@ -184,10 +184,14 @@ async def generate_text(request: Request):
|
|
184 |
print("POST Request:")
|
185 |
|
186 |
# Check length of input, if it is greater than 10 tokens, the text is sent off to a summarizer to generate:
|
187 |
-
|
188 |
-
|
189 |
-
|
190 |
-
|
|
|
|
|
|
|
|
|
191 |
# return one example
|
192 |
generated_text = generated_text[0]["generated_text"]
|
193 |
|
|
|
184 |
print("POST Request:")
|
185 |
|
186 |
# Check length of input, if it is greater than 10 tokens, the text is sent off to a summarizer to generate:
|
187 |
+
try:
|
188 |
+
generated_text = generative.generate_account_text(
|
189 |
+
prompt=data["text"], model_dir=os.path.join(models_path, data["account"])
|
190 |
+
)
|
191 |
+
logging.info("INFO: Successfully generate text from model.")
|
192 |
+
except Exception as e:
|
193 |
+
logging.error(f"Error generating text: {e}")
|
194 |
+
return {"error": "Error generating text"}
|
195 |
# return one example
|
196 |
generated_text = generated_text[0]["generated_text"]
|
197 |
|