Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def generate(prompt):
|
|
15 |
input_ids = tokenizer.encode(prompt, return_tensors="pt")
|
16 |
#output_ids = model.generate(input_ids, max_length=800, num_beams=20, early_stopping=True)
|
17 |
complete_str=""
|
18 |
-
for i in range(0,
|
19 |
outputs = model.generate(input_ids, do_sample=True, max_new_tokens=2500,early_stopping=False)
|
20 |
inp = tokenizer.decode(outputs[0])
|
21 |
complete_str=complete_str+inp+'\n'+"------"+'\n'
|
|
|
15 |
input_ids = tokenizer.encode(prompt, return_tensors="pt")
|
16 |
#output_ids = model.generate(input_ids, max_length=800, num_beams=20, early_stopping=True)
|
17 |
complete_str=""
|
18 |
+
for i in range(0, 1):
|
19 |
outputs = model.generate(input_ids, do_sample=True, max_new_tokens=2500,early_stopping=False)
|
20 |
inp = tokenizer.decode(outputs[0])
|
21 |
complete_str=complete_str+inp+'\n'+"------"+'\n'
|