pratham0203 commited on
Commit
bef26ba
1 Parent(s): ee186c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ app=FastAPI()
8
  # Use a pipeline as a high-level helper
9
 
10
 
11
- pipe = pipeline("text-generation", model="Qwen/CodeQwen1.5-7B")
12
 
13
 
14
  @app.get("/")
 
8
  # Use a pipeline as a high-level helper
9
 
10
 
11
+ pipe = pipeline("text2text-generation", model="Salesforce/codet5-small")
12
 
13
 
14
  @app.get("/")