nchen909 commited on
Commit
60da28d
1 Parent(s): a7c35df

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -6,7 +6,7 @@
6
 
7
  # app = FastAPI()
8
 
9
- # pipe_flan = pipeline("text2text-generation", model="FreedomIntelligence/Apollo-1.8B",trust_remote_code=True)
10
 
11
  # @app.get("/infer_t5")
12
  # def t5(input):
 
6
 
7
  # app = FastAPI()
8
 
9
+ # pipe_flan = pipeline("text2text-generation", model="FreedomIntelligence/Apollo-2B",trust_remote_code=True)
10
 
11
  # @app.get("/infer_t5")
12
  # def t5(input):