vitaliy-sharandin commited on
Commit
641b8c5
1 Parent(s): c3438c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ model = AutoModelForCausalLM.from_pretrained(
7
  device_map = {"": 0}
8
  )
9
 
10
- pipe = pipeline('text-generation', model=model)
11
 
12
  def generate_text(prompt):
13
  result = pipe(prompt, max_length=200)
 
7
  device_map = {"": 0}
8
  )
9
 
10
+ pipe = pipeline('text-generation', model=model, do_sample=True, temperature=0.7)
11
 
12
  def generate_text(prompt):
13
  result = pipe(prompt, max_length=200)