Gyr0MAN commited on
Commit
b8b1a88
Β·
verified Β·
1 Parent(s): 7802229

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def response(instruction, history, inputText):
19
  ### Response:"""
20
 
21
  input_ids = tokenizer(inp, return_tensors='pt').to("cuda")
22
- beam_output = model.generate(**input_ids)
23
  outputText = tokenizer.decode(beam_output[0], skip_special_token=True)
24
  #output = output.replace(inp,"").replace("<s>","").replace("</s>","")
25
 
 
19
  ### Response:"""
20
 
21
  input_ids = tokenizer(inp, return_tensors='pt').to("cuda")
22
+ beam_output = model.generate(**input_ids, max_new_tokens=300)
23
  outputText = tokenizer.decode(beam_output[0], skip_special_token=True)
24
  #output = output.replace(inp,"").replace("<s>","").replace("</s>","")
25