Update app.py

app.py

@@ -56,7 +56,7 @@ def detect(image):
     size = max(image.size)
     new_im = Image.new('RGB', (size, size), color = 0) # Create a squared black image
     new_im.paste(image)
-    new_ im = Image.fromarray(new_im.astype(np.uint8))
+    new_im = Image.fromarray(new_im.astype(np.uint8))
     
     with torch.no_grad():
   
@@ -74,6 +74,8 @@ def detect(image):
 
     pred_dress = dress_dict[pred_dress_cat]
 
+    mean_features = loaded_dict[pred_dress]
+
     pred_dress_s = f"Predicted Dress Category:  {pred_dress}"
 
     cosine_sim = cosine_similarity(dino_numpy.reshape(1, -1), mean_features.reshape(1, -1)).item()