Update app.py

app.py

@@ -5,8 +5,9 @@ from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 classifier = pipeline("image-classification", model="google/vit-base-patch16-224")
 
 # Initialize the tokenizer and model for the generative text (GPT-like model)
-tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")  # Example model, replace with your choice
-model = AutoModelForCausalLM.from_pretrained("facebook/bart-large-cnn")
+model_name = "facebook/bart-large-cnn"  # Example BART model for demonstration
+tokenizer = BartTokenizer.from_pretrained(model_name)
+model = BartForConditionalGeneration.from_pretrained(model_name)
 
 def generate_tweet(label):
     # Generate a promotional tweet using a GPT-like model