sharmavaruncs commited on
Commit
b804250
·
1 Parent(s): 200ef1d

replaced temperature values

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -297,7 +297,7 @@ def GenerateText(emo,gpt_tokenizer,gpt_model):
297
  generated = gpt_tokenizer(prompt, return_tensors="pt").input_ids
298
 
299
  sample_outputs = gpt_model.generate(generated, do_sample=True, top_k=50,
300
- max_length=30, top_p=0.95, temperature=1.9, num_return_sequences=10)#,no_repeat_ngram_size=1)
301
 
302
  # Extract and split the generated text into words
303
  outputs = set([gpt_tokenizer.decode(sample_output, skip_special_tokens=True).split(':')[-1] for sample_output in sample_outputs])
 
297
  generated = gpt_tokenizer(prompt, return_tensors="pt").input_ids
298
 
299
  sample_outputs = gpt_model.generate(generated, do_sample=True, top_k=50,
300
+ max_length=30, top_p=0.95, temperature=1.1, num_return_sequences=10)#,no_repeat_ngram_size=1)
301
 
302
  # Extract and split the generated text into words
303
  outputs = set([gpt_tokenizer.decode(sample_output, skip_special_tokens=True).split(':')[-1] for sample_output in sample_outputs])