MingGatsby commited on
Commit
b93204d
1 Parent(s): a7241b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -177,8 +177,12 @@ if uploaded_ct_file is not None:
177
  with tempfile.NamedTemporaryFile(delete=False, suffix=".dcm") as temp_file:
178
  temp_file.write(uploaded_ct_file.getvalue())
179
 
180
- # # Apply evaluation transforms to the DICOM image for model prediction
181
- # image_tensor = eval_transforms(temp_file.name).unsqueeze(0).to(device)
 
 
 
 
182
 
183
  # # Predict
184
  # with torch.no_grad():
 
177
  with tempfile.NamedTemporaryFile(delete=False, suffix=".dcm") as temp_file:
178
  temp_file.write(uploaded_ct_file.getvalue())
179
 
180
+ print(tempfile.name)
181
+
182
+ # Apply evaluation transforms to the DICOM image for model prediction
183
+ image_tensor = eval_transforms(temp_file.name).unsqueeze(0).to(device)
184
+
185
+ print(image_tensor)
186
 
187
  # # Predict
188
  # with torch.no_grad():