Detsutut commited on
Commit
3380739
1 Parent(s): 86794d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def generate_text(input_text, max_new_tokens=512, temperature=1, system_prompt="
52
  )
53
  generated_text = output[0]['generated_text']
54
 
55
- if generated_text[-1] not in [".","!","?","\n"] and generated_text[-4:-1] != "</s>:
56
  generated_text = generated_text + " [...]"
57
 
58
  split_tentative = generated_text.split("### Risposta:")
 
52
  )
53
  generated_text = output[0]['generated_text']
54
 
55
+ if generated_text[-1] not in [".","!","?","\n"] and generated_text[-4:-1] != "</s>":
56
  generated_text = generated_text + " [...]"
57
 
58
  split_tentative = generated_text.split("### Risposta:")