Spaces:

LouiSum
/

web-demo

Sleeping

LouiSum commited on Aug 14, 2023

Commit

d9691cb

1 Parent(s): 0831077

fix

Files changed (1) hide show

main.py CHANGED Viewed

@@ -1,6 +1,7 @@
-from fastapi import FastAPI, FileResponse
 from transformers import pipeline
 from fastapi.staticfiles import StaticFiles
 pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")

+from fastapi import FastAPI
 from transformers import pipeline
 from fastapi.staticfiles import StaticFiles
+from starlette.responses import FileResponse
 pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")