Abhlash commited on
Commit
335c7a7
·
verified ·
1 Parent(s): 3780ef6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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