lu-ny commited on
Commit
ffd92cf
1 Parent(s): 7442f0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -131,15 +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
- 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
 
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