Update app.py

app.py

@@ -59,8 +59,11 @@ session = None
 # Inference function
 
 @spaces.GPU()
-def predict(image):
+def predict(file):
 
+    image = cv2.imread(file.name)  # Use the file's path from Gradio
+
+    
     global session
 
     if session is None:
@@ -91,7 +94,7 @@ def predict(image):
 
 # Gradio interface
 interface = gr.Interface(fn=predict, 
-                         inputs=gr.Image(type="numpy", label="Upload an Image"),
+                         inputs=gr.File(label="Upload an Image"),  # Change to File input
                          outputs=gr.Label(num_top_classes=1, label="Predicted Class"),
                          title="ONNX Model Image Classification",
                          description="Upload an image and get the predicted class using the ONNX model.",