rsaketh02 commited on
Commit
5d749b3
1 Parent(s): 82f7aee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -5,7 +5,10 @@ from transformers import pipeline
5
  app=FastAPI()
6
 
7
 
8
- pipe = pipeline("text-generation", model="google/gemma-2b")
 
 
 
9
 
10
 
11
  @app.get("/")
 
5
  app=FastAPI()
6
 
7
 
8
+ # Use a pipeline as a high-level helper
9
+ from transformers import pipeline
10
+
11
+ pipe = pipeline("text2text-generation", model="google/flan-t5-xl")
12
 
13
 
14
  @app.get("/")