aditi2222 commited on
Commit
e3b8de8
·
1 Parent(s): 6c2f23d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def tokenize_data(text):
23
 
24
  def generate_answers(text):
25
  inputs = tokenize_data(text)
26
- results= model.generate(input_ids= inputs['input_ids'].to(device), attention_mask=inputs['attention_mask'].to(device), do_sample=True,
27
  max_length=64,
28
  top_k=120,
29
  top_p=0.98,
 
23
 
24
  def generate_answers(text):
25
  inputs = tokenize_data(text)
26
+ results= model.generate(input_ids= inputs['input_ids'], attention_mask=inputs['attention_mask'], do_sample=True,
27
  max_length=64,
28
  top_k=120,
29
  top_p=0.98,