Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -38,6 +38,7 @@ def generate_text(prompt, max_length, temperature, category):
|
|
38 |
do_sample=True
|
39 |
)
|
40 |
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
|
|
41 |
print(generated_text)
|
42 |
return generated_text
|
43 |
|
|
|
38 |
do_sample=True
|
39 |
)
|
40 |
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
41 |
+
generated_text = generated_text.replace(category_prompt, "")
|
42 |
print(generated_text)
|
43 |
return generated_text
|
44 |
|