commit

app.py

@@ -18,7 +18,7 @@ def generate_caption_and_analyze_emotions(image):
     decoded_caption = caption_processor.decode(caption[0], skip_special_tokens=True)
 
     # Load the emotion analysis model and tokenizer
-    emotion_model_name = "bhadresh-savani/distilbert-base-uncased-emotion"
+    emotion_model_name = "SamLowe/roberta-base-go_emotions"
     emotion_classifier = pipeline(model=emotion_model_name)
 
     results = emotion_classifier(decoded_caption)