Update app.py

app.py

@@ -174,18 +174,16 @@ async def get_sentiment(username: str) -> Dict[str, Dict[str, float]]:
 
 
 @app.post("/api/generate")
-# async def generate_text(account: str, text: str):
 async def generate_text(request: Request):
-    print("*"*50)
+    print("*" * 50)
     data = await request.json()
-    print("*"*50)
+    print("*" * 50)
     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'])
+        prompt=data["text"], model_dir=os.path.join(models_path, data["account"])
     )
     # return one example
     generated_text = generated_text[0]["generated_text"]
@@ -202,7 +200,6 @@ async def generate_text(request: Request):
             unique_sentences.add(sentence)
     final_text = " ".join(non_duplicate_sentences[:-1])
 
-
     return {"generated_text": final_text}
 
 @app.get("/examples1")