fix
Browse files
main.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
-
from fastapi import FastAPI
|
2 |
from transformers import pipeline
|
3 |
from fastapi.staticfiles import StaticFiles
|
|
|
4 |
|
5 |
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
|
6 |
|
|
|
1 |
+
from fastapi import FastAPI
|
2 |
from transformers import pipeline
|
3 |
from fastapi.staticfiles import StaticFiles
|
4 |
+
from starlette.responses import FileResponse
|
5 |
|
6 |
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
|
7 |
|