Spaces:
Sleeping
Sleeping
Upload mlplo/api.py with huggingface_hub
Browse files- 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-
|
| 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 |
|