Update app.py

app.py

@@ -1,14 +1,11 @@
 from fastapi import FastAPI
 from transformers import pipeline
-from huggingface_hub import login
-
 
 ## create a new FASTAPI app instance
 app=FastAPI()
 
-
 # Initialize the text generation pipeline
-pipe = pipeline("text-generation", model="google/flan-t5-small")
+pipe = pipeline("text2text-generation", model="google/flan-t5-small")
 
 
 @app.get("/")
@@ -25,5 +22,3 @@ def generate(text:str):
 
     ## return the generate text in Json reposne
     return {"output":output[0]['generated_text']}
-
-