pratik-aivantage commited on
Commit
ac328ee
·
verified ·
1 Parent(s): 756d547

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
32
 
33
  def generate_answer(question):
34
  inputs = tokenizer.encode("Question: " + question, return_tensors="pt")
35
- outputs = model.generate(inputs, max_length=100, num_return_sequences=1, early_stopping=True)
36
  answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
37
  return answer
38
 
 
32
 
33
  def generate_answer(question):
34
  inputs = tokenizer.encode("Question: " + question, return_tensors="pt")
35
+ outputs = model.generate(inputs, max_length=100, num_return_sequences=1, early_stopping=True, do_sample=True)
36
  answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
37
  return answer
38