Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ def calculate(request: CalculationRequest):
|
|
| 35 |
# Example endpoint using Hugging Face model
|
| 36 |
@app.post("/generate")
|
| 37 |
def generate_text(prompt: str):
|
| 38 |
-
generated = model(prompt, max_length=50)
|
| 39 |
return {"generated_text": generated[0]['generated_text']}
|
| 40 |
|
| 41 |
# New endpoint for testing math operations
|
|
|
|
| 35 |
# Example endpoint using Hugging Face model
|
| 36 |
@app.post("/generate")
|
| 37 |
def generate_text(prompt: str):
|
| 38 |
+
generated = model(prompt, max_length=50, clean_up_tokenization_spaces=True)
|
| 39 |
return {"generated_text": generated[0]['generated_text']}
|
| 40 |
|
| 41 |
# New endpoint for testing math operations
|