Update app.py

app.py

@@ -38,6 +38,7 @@ def generate_text(prompt, max_length, temperature, category):
         do_sample=True
     )
     generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    generated_text = generated_text.replace(category_prompt, "")
     print(generated_text)
     return generated_text