Update app.py

app.py

@@ -11,9 +11,9 @@ model = BartForConditionalGeneration.from_pretrained(model_name)
 
 def generate_tweet(label):
     # Generate a promotional tweet using a GPT-like model
-    prompt = f"Write a tweet about {label}:"
+    prompt = f"{label}:"
     inputs = tokenizer.encode(prompt, return_tensors="pt")
-    outputs = model.generate(inputs, max_length=280, num_return_sequences=1)
+    outputs = model.generate(inputs, max_length=280, num_return_sequences=1, no_repeat_ngram_size=2)
     tweet = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return tweet