improvements

app.py

@@ -16,8 +16,11 @@ transform = transforms.Compose([
 ])
 
 # Function to load and preprocess an image
-def load_image(image_path):
-    image = Image.open(image_path).convert('RGB')
+def load_image(image):
+    if isinstance(image, str):  # If image is a file path
+        image = Image.open(image).convert('RGB')
+    elif isinstance(image, np.ndarray):  # If image is a numpy array
+        image = Image.fromarray(image.astype('uint8'), 'RGB')
     image = transform(image).unsqueeze(0)  # Add batch dimension
     return image
 
@@ -34,12 +37,8 @@ def predict_image(image):
     if image is None:
         return "No image provided", None
     
-    # Convert the image to PIL if it's not a filepath
-    if isinstance(image, np.ndarray):
-        image = Image.fromarray(image.astype('uint8'), 'RGB')
-    
-    # Load and transform the image from the file path
-    img_tensor = transform(image).unsqueeze(0)
+    # Load and transform the image
+    img_tensor = load_image(image)
     
     # Perform the prediction
     with torch.no_grad():
@@ -72,7 +71,7 @@ def predict_image(image):
 iface = gr.Interface(
     fn=predict_image,
     inputs=[
-        gr.Image(source="upload", type="pil", tool="editor", label="Upload an image or take a photo"),
+        gr.Image(type="pil", tool="editor", label="Upload an image or take a photo"),
         gr.Image(source="webcam", type="pil", tool="editor", label="Take a photo")
     ],
     outputs=[