lemonaddie commited on
Commit
77fa61c
1 Parent(s): b8f574d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -22
app.py CHANGED
@@ -276,28 +276,7 @@ def main():
276
 
277
  pipe = pipe.to(device)
278
 
279
- input_image = Image.open('files/bee.jpg')
280
-
281
- pipe_out = pipe(
282
- input_image,
283
- denoising_steps=5,
284
- ensemble_size=1,
285
- processing_res=768,
286
- batch_size=0,
287
- guidance_scale=3,
288
- domain="indoor",
289
- show_progress_bar=True,
290
- )
291
-
292
- depth_pred: np.ndarray = pipe_out.depth_np
293
- depth_colored: Image.Image = pipe_out.depth_colored
294
- normal_pred: np.ndarray = pipe_out.normal_np
295
- normal_colored: Image.Image = pipe_out.normal_colored
296
-
297
- print(depth_pred.shape)
298
- print(np.unique(depth_pred))
299
-
300
- # run_demo_server(pipe)
301
 
302
 
303
  if __name__ == "__main__":
 
276
 
277
  pipe = pipe.to(device)
278
 
279
+ run_demo_server(pipe)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
280
 
281
 
282
  if __name__ == "__main__":