Kryko7 commited on
Commit
c44e55e
1 Parent(s): 023ea2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,7 +1,12 @@
1
  from fastapi import FastAPI
2
  from transformers import pipeline
3
 
4
- app = FastAPI()
 
 
 
 
 
5
 
6
 
7
  pipe = pipeline("text-generation", model="google/flan-t5-small")
 
1
  from fastapi import FastAPI
2
  from transformers import pipeline
3
 
4
+ app = FastAPI(
5
+ version="1.0",
6
+ openapi_url="/openapi.json",
7
+ docs_url="/swagger",
8
+ redoc_url=None,
9
+ )
10
 
11
 
12
  pipe = pipeline("text-generation", model="google/flan-t5-small")