Bunpheng commited on
Commit
447f9ff
1 Parent(s): b3ff035

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,9 +4,9 @@ import os
4
 
5
  app = FastAPI()
6
 
7
- pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
8
 
9
- @app.get("/t5")
10
  def t5(input):
11
  output = pipe_flan(input)
12
  return {"output": output[0]["generated_text"]}
 
4
 
5
  app = FastAPI()
6
 
7
+ pipe_flan = pipeline("text2text-generation", model="mistralai/Mistral-7B-v0.3")
8
 
9
+ @app.get("/mistral")
10
  def t5(input):
11
  output = pipe_flan(input)
12
  return {"output": output[0]["generated_text"]}