egosumkira commited on
Commit
d6c38d2
·
1 Parent(s): 265eae6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ story = pipeline(
16
  def generate(tags_text, temp, n_beams, max_len):
17
  tags = tags_text.split(", ")
18
  prefix = f"~^{'^'.join(tags)}~@"
19
- g_text = story(prefix, temperature=temp, repetition_penalty=7.0, num_beams=n_beams, max_length=max_len)[0]['generated_text']
20
  return g_text[g_text.find("@") + 1:]
21
 
22
 
 
16
  def generate(tags_text, temp, n_beams, max_len):
17
  tags = tags_text.split(", ")
18
  prefix = f"~^{'^'.join(tags)}~@"
19
+ g_text = story(prefix, temperature=float(temp), repetition_penalty=7.0, num_beams=int(n_beams), max_length=int(max_len))[0]['generated_text']
20
  return g_text[g_text.find("@") + 1:]
21
 
22