shkna1368 commited on
Commit
9c8c7b8
1 Parent(s): d1d24e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, 2):
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'