Update app.py

app.py

@@ -17,8 +17,9 @@ with torch.no_grad():
     attribute_embeddings = model.encode_text(clip.tokenize(attributes).to(device))
 
 def predict_apparel_and_attributes(image):
-    # Process image and compute its embedding
-    image_input = preprocess(image).unsqueeze(0).to(device)
+    pil_image = Image.fromarray((image * 255).astype(np.uint8))
+    image_input = preprocess(pil_image).unsqueeze(0).to(device)
+    
     with torch.no_grad():
         image_embedding = model.encode_image(image_input)