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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -134,12 +134,12 @@ def generate_novel_title_and_summary(genres, tones, themes, temperature=1.5, max
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
 
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