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).unsqueeze(0)  # Add batch dimension
+        #image = Image.fromarray(image_path)
+        input_tensor = transform(image_path).unsqueeze(0)  # Add batch dimension
         
         
         outputs = model(input_tensor)
@@ -32,4 +32,4 @@ def predict(image_path):
         probability_dict = {}
         for i, prob in enumerate(probabilities):
           probability_dict[labels[i]] = prob.item()
-        return predicted_label
+        return probability_dict