Update app.py
Browse files
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 |
|