Update app.py

app.py

@@ -5,8 +5,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 app = FastAPI()
 app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"], allow_credentials=False,
+    CORSMiddleware, allow_origins=["*"], allow_credentials=False,
     allow_methods=["*"], allow_headers=["*"]
 )
 
@@ -30,4 +29,5 @@ async def generate_story(req: PromptRequest):
         repetition_penalty=1.2,
         do_sample=True
     )
-    return {"story": tokenizer.decode(outputs[0], skip_special_tokens=True)}
+    story = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return {"story": story}