Spaces:
Runtime error
Runtime error
Commit
·
d6c38d2
1
Parent(s):
265eae6
Update app.py
Browse files
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 |
|