Ld75 commited on
Commit
96dd1aa
·
1 Parent(s): 5f1ccf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -6,19 +6,19 @@ from fastapi.staticfiles import StaticFiles
6
  from fastapi.responses import FileResponse
7
  #from pyannote.audio import Pipeline
8
 
9
- from transformers import pipeline
10
 
11
  app = FastAPI()
12
 
13
- pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
14
 
15
  @app.get("/")
16
  def t5(input):
17
 
18
- output = pipe_flan(input)
19
  #pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
20
  #pipeline("file.wav")
21
- return {"output":"OK"+output}
22
 
23
  #app.mount("/", StaticFiles(directory="static", html=True), name="static")
24
 
 
6
  from fastapi.responses import FileResponse
7
  #from pyannote.audio import Pipeline
8
 
9
+ #from transformers import pipeline
10
 
11
  app = FastAPI()
12
 
13
+ #pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
14
 
15
  @app.get("/")
16
  def t5(input):
17
 
18
+ #output = pipe_flan(input)
19
  #pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
20
  #pipeline("file.wav")
21
+ return {"output":"OK"}
22
 
23
  #app.mount("/", StaticFiles(directory="static", html=True), name="static")
24