commit

app.py

@@ -1,5 +1,8 @@
 import gradio as gr
 from transformers import BlipProcessor, BlipForConditionalGeneration, RobertaTokenizer, RobertaForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
 
 # Load the image captioning model and tokenizer
 caption_model_name = "Salesforce/blip-image-captioning-large"
@@ -8,8 +11,8 @@ caption_model = BlipForConditionalGeneration.from_pretrained(caption_model_name)
 
 # Load the emotion analysis model and tokenizer
 emotion_model_name = "SamLowe/roberta-base-go_emotions"
-emotion_tokenizer = RobertaTokenizer.from_pretrained(emotion_model_name)
-emotion_model = RobertaForSequenceClassification.from_pretrained(emotion_model_name)
+emotion_tokenizer = AutoTokenizer.from_pretrained(emotion_model_name)
+emotion_model = AutoModelForSequenceClassification.from_pretrained(emotion_model_name)
 
 def generate_caption_and_analyze_emotions(image):
     # Preprocess the image for caption generation