Adive01 commited on
Commit
f574adb
·
verified ·
1 Parent(s): 3d93a56

Upload mlplo/api.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. mlplo/api.py +1 -1
mlplo/api.py CHANGED
@@ -86,7 +86,7 @@ class SummarizeRequest(BaseModel):
86
  min_new_tokens: int = 30
87
  num_beams: int = 4
88
  length_penalty: float = 1.5 # >1 encourages longer, more complete summaries
89
- gemini_model: str = "gemini-3.0-flash"
90
  polish: bool = False # if True, run Gemini to clean up BART's output
91
 
92
 
 
86
  min_new_tokens: int = 30
87
  num_beams: int = 4
88
  length_penalty: float = 1.5 # >1 encourages longer, more complete summaries
89
+ gemini_model: str = "gemini-2.5-flash"
90
  polish: bool = False # if True, run Gemini to clean up BART's output
91
 
92