Update app.py

app.py

@@ -38,14 +38,13 @@ def detect(image):
     with torch.no_grad():
   
   # Apply transformations to the image and move it to the appropriate device
-    image_tensor = transform_image(new_im).to(device)
+        image_tensor = transform_image(new_im).to(device)
   
   # Extract features using the DinoV2 model
-    dino_embedding = dinov2_vits14(image_tensor.unsqueeze(0)).cpu()
-    dino_numpy = dinov2_vits14(image_tensor.unsqueeze(0)).cpu().numpy()
-
-    model.eval()
+        dino_embedding = dinov2_vits14(image_tensor.unsqueeze(0)).cpu()
+        dino_numpy = dinov2_vits14(image_tensor.unsqueeze(0)).cpu().numpy()
 
+    
     with torch.no_grad():
         outputs = model(dino_embedding)
         pred_dress_cat = round(torch.argmax(outputs, dim = 1).tolist()[0])