vitaliy-sharandin commited on
Commit
6035fd2
1 Parent(s): 650bd03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ tokenizer = AutoTokenizer.from_pretrained('vitaliy-sharandin/wiseai')
11
  pipe = pipeline('text-generation', model=model,tokenizer=tokenizer)
12
 
13
  def generate_text(prompt):
14
- result = pipe(prompt, max_length=200, do_sample=True, temperature=0.7, num_return_sequences=1, return_full_text=False)
15
  return result[0]['generated_text']
16
 
17
  iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
 
11
  pipe = pipeline('text-generation', model=model,tokenizer=tokenizer)
12
 
13
  def generate_text(prompt):
14
+ result = pipe(prompt, max_length=200, do_sample=True, temperature=0.9, num_return_sequences=1, return_full_text=False)
15
  return result[0]['generated_text']
16
 
17
  iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")