Spaces:
Running
on
Zero
Running
on
Zero
lemonaddie
commited on
Commit
•
d2a51cd
1
Parent(s):
9822b8a
Update models/geowizard_pipeline.py
Browse files
models/geowizard_pipeline.py
CHANGED
@@ -198,6 +198,8 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
|
|
198 |
|
199 |
normal_colored = ((normal_pred + 1)/2 * 255).astype(np.uint8)
|
200 |
normal_colored_img = Image.fromarray(normal_colored)
|
|
|
|
|
201 |
|
202 |
return DepthNormalPipelineOutput(
|
203 |
depth_np = depth_pred,
|
|
|
198 |
|
199 |
normal_colored = ((normal_pred + 1)/2 * 255).astype(np.uint8)
|
200 |
normal_colored_img = Image.fromarray(normal_colored)
|
201 |
+
|
202 |
+
self.img_embed = None
|
203 |
|
204 |
return DepthNormalPipelineOutput(
|
205 |
depth_np = depth_pred,
|