Update model/model.py

model/model.py

@@ -18,8 +18,8 @@ def predict(image_path):
         ])
     
         # Preprocess the image
-        #image = Image.fromarray(image_path)
-        input_tensor = transform(image_path).unsqueeze(0)  # Add batch dimension
+        image_tensor = torch.from_numpy(image_path).permute(2, 0, 1).float() 
+        input_tensor = transform(image_tensor).unsqueeze(0)  # Add batch dimension
         
         
         outputs = model(input_tensor)