Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -69,6 +69,7 @@ def pred_maps(images):
|
|
69 |
scaled_preds_tensor = model(images_proc.to(device))[-1]
|
70 |
preds = []
|
71 |
print('scaled_preds_tensor.sum():', scaled_preds_tensor.sum())
|
|
|
72 |
for image_shape, pred_tensor, save_path in zip(image_shapes, scaled_preds_tensor, save_paths):
|
73 |
if device == 'cuda':
|
74 |
pred_tensor = pred_tensor.cpu()
|
|
|
69 |
scaled_preds_tensor = model(images_proc.to(device))[-1]
|
70 |
preds = []
|
71 |
print('scaled_preds_tensor.sum():', scaled_preds_tensor.sum())
|
72 |
+
print('type(scaled_preds_tensor):', type(scaled_preds_tensor))
|
73 |
for image_shape, pred_tensor, save_path in zip(image_shapes, scaled_preds_tensor, save_paths):
|
74 |
if device == 'cuda':
|
75 |
pred_tensor = pred_tensor.cpu()
|