Spaces:
Sleeping
Sleeping
remove float model
Browse files
app.py
CHANGED
|
@@ -112,8 +112,8 @@ def predict_image(image_bef, image_aft, json_data_bef, json_data_aft):
|
|
| 112 |
|
| 113 |
left_map, right_map = torch.from_numpy(left_map).unsqueeze(0), torch.from_numpy(right_map).unsqueeze(0)
|
| 114 |
|
| 115 |
-
bef_image = torch.from_numpy(_get_rawimage(image_bef)).unsqueeze(1)
|
| 116 |
-
aft_image = torch.from_numpy(_get_rawimage(image_aft)).unsqueeze(1)
|
| 117 |
|
| 118 |
image_pair = torch.cat([bef_image, aft_image], 1)
|
| 119 |
|
|
|
|
| 112 |
|
| 113 |
left_map, right_map = torch.from_numpy(left_map).unsqueeze(0), torch.from_numpy(right_map).unsqueeze(0)
|
| 114 |
|
| 115 |
+
bef_image = torch.from_numpy(_get_rawimage(image_bef)).unsqueeze(1)
|
| 116 |
+
aft_image = torch.from_numpy(_get_rawimage(image_aft)).unsqueeze(1)
|
| 117 |
|
| 118 |
image_pair = torch.cat([bef_image, aft_image], 1)
|
| 119 |
|