Update app.py

app.py

@@ -286,6 +286,9 @@ if uploaded_ct_file is not None:
 
     transformed_array = eval_transforms(dicom_array)
 
+    # Convert to PyTorch tensor and move to device
+    image_tensor = transformed_array.clone().detach().unsqueeze(0).to(device)
+
     # Predict
     with torch.no_grad():
         outputs = ct_model(image_tensor).sigmoid().to("cpu").numpy()