Hugging Face's logo

Spaces:
SpacesExamples
/
fastapi_t5
Runtime error

App Files Community
4

Update main.py

#3
by nuojohnchen - opened
base: refs/heads/main
←
from: refs/pr/3
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -6,7 +6,7 @@ from transformers import pipeline
6
 
7
  app = FastAPI()
8
 
9
- pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
10
 
11
  @app.get("/infer_t5")
12
  def t5(input):
 
6
 
7
  app = FastAPI()
8
 
9
+ pipe_flan = pipeline("text2text-generation", model="FreedomIntelligence/Apollo-0.5B")
10
 
11
  @app.get("/infer_t5")
12
  def t5(input):