Upload app.py

app.py

@@ -22,6 +22,11 @@ download_file_from_google_drive(MODEL_URL, MODEL_PATH)
 model = tf.keras.models.load_model(MODEL_PATH)
 
 def predict_digit(image):
+    # If the input is a file path, open and process the image
+    if isinstance(image, str):
+        from PIL import Image
+        image = Image.open(image).convert('L')
+        image = np.array(image)
     if image.ndim == 3:
         image = image[..., 0]
     image = np.array(image).astype("float32")
@@ -32,10 +37,10 @@ def predict_digit(image):
 
 iface = gr.Interface(
     fn=predict_digit,
-    inputs=gr.Image(image_mode='L', height=28, width=28),
+    inputs=gr.Image(image_mode='L', type='filepath', height=28, width=28),
     outputs=gr.Label(num_top_classes=1),
     title="MNIST Digit Classifier",
-    description="Draw a digit (0-9) and the model will predict it."
+    description="Upload or draw a digit (0-9) and the model will predict it."
 )
 
 if __name__ == "__main__":