Update app.py

app.py

@@ -43,13 +43,13 @@ os.system("wget https://github.com/onnx/models/raw/main/vision/classification/zf
 
 ort_session = ort.InferenceSession("zfnet512-12.onnx")
 
-    
+
 def predict(path):
     img_batch = preprocess(get_image(path))
 
     outputs = ort_session.run(
         None,
-        {"data_0": img_batch.astype(np.float32)},
+        {"gpu_0/data_0": img_batch.astype(np.float32)},
     )
 
     a = np.argsort(-outputs[0].flatten())