pooyanrg commited on
Commit
cb596ec
·
1 Parent(s): ffc2578

remove float model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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).float()
116
- aft_image = torch.from_numpy(_get_rawimage(image_aft)).unsqueeze(1).float()
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