Update app.py

app.py

@@ -11,7 +11,7 @@ model = BartForConditionalGeneration.from_pretrained(model_name)
 
 def generate_tweet(label):
     # Updated to generate a promotional tweet without mentioning emails and websites
-    prompt = f"Write a tweet about {label}"
+    prompt = f"Write a short post about {label}"
     
     inputs = tokenizer.encode(prompt, return_tensors="pt")
     outputs = model.generate(inputs, max_length=280, num_return_sequences=1, no_repeat_ngram_size=2)
@@ -19,7 +19,7 @@ def generate_tweet(label):
     tweet_raw = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
     # Process the generated tweet to remove unwanted patterns (additional processing can be added if needed)
-    tweet = tweet_raw.replace("Write a tweet about", "")  # Ensure this phrase is not included
+    tweet = tweet_raw.replace("Write a short post about", "")  # Ensure this phrase is not included
     
     return tweet