Update app.py

app.py

@@ -32,7 +32,7 @@ def imaging_agent(image_path: str):
         arr = xrv.datasets.normalize(arr, 4096)
 
         # Wrap in dict for TorchXRayVision
-        sample = {"img": arr}
+        sample = {"img": arr, "lab": np.array([]), "idx": np.array([])}
 
         # Apply transforms step by step
         sample = xrv.datasets.XRayCenterCrop()(sample)