Update app.py

app.py

@@ -5,20 +5,20 @@ from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
 #from pyannote.audio import Pipeline
-  
-#from transformers import pipeline
+ 
+from transformers import pipeline
 
 app = FastAPI()
 
-#pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
 
-@app.get("/healthcheck")
+@app.get("/")
 def t5(input):
     
-    #output = pipe_flan(input)
+    output = pipe_flan(input)
     #pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
     #pipeline("file.wav")
-    return {"output":"OK"}
+    return {"output":"OK"+output}
 
 #app.mount("/", StaticFiles(directory="static", html=True), name="static")