Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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()
|