Update main.py

main.py

@@ -3,6 +3,15 @@ from fastapi.responses import StreamingResponse
 import os
 import io
 
+from transformers import pipeline
+
+
+pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+
+
+output = pipe_flan("hi")
+print(output[0]["generated_text"])
+
 temp = open("model/t.txt","r")
 
 app = FastAPI()