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 |
|