Upload app.py

app.py

@@ -1,7 +1,11 @@
 from fastapi import FastAPI
+from transformers import pipeline
 
 app = FastAPI()
 
+# Initialize the text generation pipeline
+pipe = pipeline("text2text-generation", model="google/flan-t5-small")
+
 @app.get("/")
 async def root():
     return {"message": "Hello World"}