yasserrmd commited on
Commit
ad72b3e
1 Parent(s): 5cdf426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -34,8 +34,7 @@ def generate_text(prompt, max_length, temperature, category):
34
  outputs = model.generate(
35
  **inputs,
36
  max_length=max_length,
37
- temperature=temperature,
38
- do_sample=True
39
  )
40
  generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
41
  generated_text = generated_text.replace(category_prompt, "")
 
34
  outputs = model.generate(
35
  **inputs,
36
  max_length=max_length,
37
+ temperature=temperature
 
38
  )
39
  generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
40
  generated_text = generated_text.replace(category_prompt, "")