DeepVen commited on
Commit
1af91d4
1 Parent(s): 78b1acd

Upload main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -14,7 +14,7 @@ embeddings.load('index')
14
  # Create extractor instance
15
  extractor = Extractor(embeddings, "google/flan-t5-base")
16
 
17
- pipe = pipeline("text2text-generation", model="google/flan-t5-xxl")
18
 
19
 
20
  @app.get("/generate")
 
14
  # Create extractor instance
15
  extractor = Extractor(embeddings, "google/flan-t5-base")
16
 
17
+ pipe = pipeline("text2text-generation", model="google/flan-t5-large")
18
 
19
 
20
  @app.get("/generate")