Update app.py

app.py

@@ -40,10 +40,10 @@ def process_image(image):
         clothing_text_features /= clothing_text_features.norm(dim=-1, keepdim=True)
 
         # Calculate probabilities
-        type_text_probs = (100.0 * image_features @ type_text_features.T).softmax(dim=-1)
-        scene_text_probs = (100.0 * image_features @ scene_text_features.T).softmax(dim=-1)
-        expression_text_probs = (100.0 * image_features @ expression_text_features.T).softmax(dim=-1)
-        clothing_text_probs = (100.0 * image_features @ clothing_text_features.T).softmax(dim=-1)
+        type_text_probs = (100.0 * image_features @ type_text_features.T).sigmoid(dim=-1)
+        scene_text_probs = (100.0 * image_features @ scene_text_features.T).sigmoid(dim=-1)
+        expression_text_probs = (100.0 * image_features @ expression_text_features.T).sigmoid(dim=-1)
+        clothing_text_probs = (100.0 * image_features @ clothing_text_features.T).sigmoid(dim=-1)
 
     # Convert to dictionaries
     type_results = {label: float(type_text_probs[0][i]) for i, label in enumerate(type_labels)}