Tsmith2024 commited on
Commit
b45f032
·
verified ·
1 Parent(s): a6bdc3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -178,6 +178,7 @@ with tempfile.TemporaryDirectory() as tmpdir:
178
  def cb_generate(image, depth, scale_factor, sd_version):
179
  depth_tensor = torch.tensor(depth).unsqueeze(0).unsqueeze(0).float()
180
  norm_disp = normalize_disp(depth_tensor.cuda())
 
181
 
182
  genstereo = get_genstereo_model(sd_version)
183
  fusion_model = get_fusion_model()
 
178
  def cb_generate(image, depth, scale_factor, sd_version):
179
  depth_tensor = torch.tensor(depth).unsqueeze(0).unsqueeze(0).float()
180
  norm_disp = normalize_disp(depth_tensor.cuda())
181
+ disp = norm_disp * scale_factor / 100 * IMAGE_SIZE
182
 
183
  genstereo = get_genstereo_model(sd_version)
184
  fusion_model = get_fusion_model()