okeanos commited on
Commit
25a8d0a
1 Parent(s): d7744cf
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -17,8 +17,6 @@ def generate_text(prompt):
17
  prompt = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
18
  outputs = generator(
19
  prompt,
20
- do_sample=True
21
- return_full_text=False
22
  )
23
  # Extract the generated text and return it
24
  generated_text = outputs[0]['generated_text']
 
17
  prompt = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
18
  outputs = generator(
19
  prompt,
 
 
20
  )
21
  # Extract the generated text and return it
22
  generated_text = outputs[0]['generated_text']