update

app.py

@@ -10,38 +10,35 @@ from transformers import AutoProcessor, AutoModelForImageClassification
 # Load environment variables from .env file
 load_dotenv()
 
-# Set up the Hugging Face API for emotion detection
-emotion_model_url = "https://api-inference.huggingface.co/models/trpakov/vit-face-expression"
-headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_API_KEY')}"}
-
 # Set up OpenAI API key
 openai.api_key = os.getenv('OPENAI_API_KEY')
 
+# Load the processor and model for facial expression recognition
+processor = AutoProcessor.from_pretrained("trpakov/vit-face-expression")
+model = AutoModelForImageClassification.from_pretrained("trpakov/vit-face-expression")
+
 # Function to query the facial expression recognition model
 def query_emotion(image):
-    # Load the processor and model
-    processor = AutoProcessor.from_pretrained("trpakov/vit-face-expression")
-    model = AutoModelForImageClassification.from_pretrained("trpakov/vit-face-expression")
-
     # Preprocess the image
     inputs = processor(images=image, return_tensors="pt")
 
     # Perform inference
     with torch.no_grad():
         outputs = model(**inputs)
-    
-    # Get predicted class index
+
+    # Get predicted class index (the class with the highest logit)
     logits = outputs.logits
     predicted_class_idx = torch.argmax(logits, dim=-1).item()
 
-    # Get the predicted label
-    predicted_label = processor.decode(predicted_class_idx)
+    # Retrieve the label names from the model
+    label_names = model.config.id2label  # Mapping of indices to emotion labels
+    predicted_label = label_names[predicted_class_idx]  # Get the predicted label
+
     return predicted_label
 
 # Function to generate a response using OpenAI based on detected emotion
 def generate_text_based_on_mood(emotion):
     try:
-        # Create a dynamic prompt based on the detected emotion
         prompt = f"Generate a light-hearted joke or motivational message for someone who is feeling {emotion}."
         
         # Call OpenAI's API using GPT-4