pablo commited on
Commit
0de5570
·
1 Parent(s): efcd5c0

CUANTOS ERRORES MAS LOCO

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -94,8 +94,8 @@ def predict(dict, depth, prompt="", negative_prompt="", guidance_scale=7.5, step
94
  output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, depth_image=depth_image, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
95
 
96
  depth_out = np.array(output.depth[0])
97
-
98
- output_depth_vis = (depth_out - np.min(output.depth[0])) / (np.max(depth_out) - np.min(depth_out)) * 255
99
  output_depth_vis = output_depth_vis.astype("uint8")
100
  output_depth = Image.fromarray(output_depth_vis)
101
 
 
94
  output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, depth_image=depth_image, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
95
 
96
  depth_out = np.array(output.depth[0])
97
+
98
+ output_depth_vis = (depth_out - np.min(depth_out)) / (np.max(depth_out) - np.min(depth_out)) * 255
99
  output_depth_vis = output_depth_vis.astype("uint8")
100
  output_depth = Image.fromarray(output_depth_vis)
101