Update app.py

app.py

@@ -177,8 +177,21 @@ if uploaded_ct_file is not None:
     with tempfile.NamedTemporaryFile(delete=False, suffix=".dcm") as temp_file:
         temp_file.write(uploaded_ct_file.getvalue())
 
-        # Apply evaluation transforms to the DICOM image for model prediction
-        image_tensor = eval_transforms(temp_file.name).unsqueeze(0).to(device)
+        # Debugging section to isolate the RuntimeError issue
+        try:
+            # Check if the temporary DICOM file is accessible and properly written
+            if not os.path.exists(temp_file.name) or os.path.getsize(temp_file.name) == 0:
+                print("Debugging: Temporary DICOM file is either missing or empty.")
+        
+            # Attempt to apply the evaluation transforms to the DICOM image
+            image_tensor = eval_transforms(temp_file.name).unsqueeze(0).to(device)
+        
+        except Exception as e:
+            print(f"Debugging: Exception caught while applying transform: {e}")
+            raise
+
+        # # Apply evaluation transforms to the DICOM image for model prediction
+        # image_tensor = eval_transforms(temp_file.name).unsqueeze(0).to(device)
 
         # Predict
         with torch.no_grad():