lu-ny commited on
Commit
982e372
·
1 Parent(s): e52eec2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -131,9 +131,15 @@ def generate_novel_title_and_summary(genres, tones, themes, temperature=1.5, max
131
  data = {
132
  "inputs": prompt,
133
  "options": {
134
- "temperature": temperature,
135
- "do_sample": True,
136
- "use_cache": False
 
 
 
 
 
 
137
  }
138
  }
139
  # You can get an Inference API for FREE
 
131
  data = {
132
  "inputs": prompt,
133
  "options": {
134
+ temperature: temperature,
135
+ do_sample: True,
136
+ use_cache: False,
137
+ max_new_tokens=250,
138
+ min_new_tokens=100,
139
+ do_sample = True,
140
+ use_cache = False,
141
+ max_p = 0.95,
142
+ repetition_penalty = 1.15,
143
  }
144
  }
145
  # You can get an Inference API for FREE