zmbfeng commited on
Commit
3759c66
·
verified ·
1 Parent(s): 1b14365

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def create_response_untethered_paraphrased(input_str,
49
  input_ids = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors="pt")
50
  #output_ids = fine_tuned_model.generate(input_ids,do_sample=True, max_length=100, temperature=0.2, top_p=0.9, repetition_penalty=1.5,num_return_sequences=6)
51
  #output_ids = untethered_paraphrased_model.generate(input_ids,do_sample=do_sample, max_length=100, temperature=temperature, top_p=top_p, top_k=top_k, repetition_penalty=repetition_penalty,num_return_sequences=num_return_sequences, num_beams = num_beams)
52
- output_ids = untethered_paraphrased_model.generate(input_ids,do_sample=True, max_length=100, temperature=temperature, top_p=top_p, repetition_penalty=repetition_penalty,num_return_sequences=num_return_sequences,num_beams=num_beams)
53
  outputs = ""
54
  for output_id in output_ids:
55
  output = tokenizer.decode(output_id, skip_special_tokens=True)
 
49
  input_ids = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors="pt")
50
  #output_ids = fine_tuned_model.generate(input_ids,do_sample=True, max_length=100, temperature=0.2, top_p=0.9, repetition_penalty=1.5,num_return_sequences=6)
51
  #output_ids = untethered_paraphrased_model.generate(input_ids,do_sample=do_sample, max_length=100, temperature=temperature, top_p=top_p, top_k=top_k, repetition_penalty=repetition_penalty,num_return_sequences=num_return_sequences, num_beams = num_beams)
52
+ output_ids = untethered_paraphrased_model.generate(input_ids,do_sample=True, max_length=100, temperature=temperature, top_p=top_p, top_k=top_k, repetition_penalty=repetition_penalty,num_return_sequences=num_return_sequences)
53
  outputs = ""
54
  for output_id in output_ids:
55
  output = tokenizer.decode(output_id, skip_special_tokens=True)