chenguittiMaroua commited on
Commit
a65b358
·
verified ·
1 Parent(s): 4e61109

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -133,8 +133,8 @@ def get_summarizer():
133
 
134
  MODEL_CHOICES = [
135
  "cmarkea/flan-t5-base-fr", # Best for French
136
- "google/flan-t5-base",
137
- "moussaKam/barthez-orangesum-abstract" # French summarization
138
  ]
139
 
140
  qa_pipeline = None
 
133
 
134
  MODEL_CHOICES = [
135
  "cmarkea/flan-t5-base-fr", # Best for French
136
+ "moussaKam/barthez-orangesum-abstract", # French summarization
137
+ "google/flan-t5-xl" # Higher quality fallback
138
  ]
139
 
140
  qa_pipeline = None