Update app.py

app.py

@@ -24,7 +24,7 @@ def get_mood_from_image(image: Image.Image):
     inputs = processor(text=text_inputs, images=image, return_tensors="pt", padding=True)
     
     # Run the model
-    logits = model(**inputs).logits
+    logits = model(**inputs).logits_per_image
     probs = logits.softmax(dim=-1).tolist()
 
     # Calculate the scores for each mood