Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py
 import torch
 import torch.nn as nn
 import torchvision.transforms as transforms
@@ -46,7 +47,7 @@ def predict(image):
 # Create the Gradio interface
 iface = gr.Interface(
     fn=predict, 
-    inputs=gr.inputs.Image(shape=(28, 28), image_mode='L', invert_colors=False), 
+    inputs=gr.Image(shape=(28, 28), image_mode='L', invert_colors=False), 
     outputs="label",
     live=True
 )