Update app.py

app.py

@@ -1,16 +1,16 @@
-    from transformers import pipeline
-    from fastapi import FastAPI, Request, HTTPException
+from transformers import pipeline
+from fastapi import FastAPI, Request, HTTPException
 
-    app = FastAPI()
+app = FastAPI()
     
-    generator = pipeline('text2text-generation', model='google/flan-t5-small')
+generator = pipeline('text2text-generation', model='google/flan-t5-small')
     
-    @app.post("/")
-    async def generate(request: Request):
-        try:
-            data = await request.json()
-            text = data["text"]
-            output = generator(text)
-            return {"output": output[0]["generated_text"]}
-        except Exception as e:
-             raise HTTPException(status_code=500, detail=str(e))
+@app.post("/")
+async def generate(request: Request):
+    try:
+        data = await request.json()
+        text = data["text"]
+        output = generator(text)
+        return {"output": output[0]["generated_text"]}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))