ahmadmac commited on
Commit
293e1b9
·
verified ·
1 Parent(s): 8de3193

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def generate_text(prompt, max_length=50, num_return_sequences=1, temperature=0.7
11
  input_ids,
12
  max_length=max_length,
13
  num_return_sequences=num_return_sequences,
14
- temperature=temperature  
15
 
16
  )
17
  return tokenizer.decode(output[0], skip_special_tokens=True)
 
11
  input_ids,
12
  max_length=max_length,
13
  num_return_sequences=num_return_sequences,
14
+ temperature=0.7 
15
 
16
  )
17
  return tokenizer.decode(output[0], skip_special_tokens=True)