bartduis commited on
Commit
6c23b9d
·
verified ·
1 Parent(s): f5bbdd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -10
app.py CHANGED
@@ -9,9 +9,7 @@ def dummy_warmup():
9
  else:
10
  print("[WARNING] CUDA not available. Skipping warmup.")
11
 
12
- if __name__ == "__main__":
13
- dummy_warmup()
14
-
15
  # import numpy as np
16
  # import gradio as gr
17
  # import rembg
@@ -198,6 +196,8 @@ if __name__ == "__main__":
198
  # rgb = output_np[:,:,:3]
199
  # return mask, rgb
200
 
 
 
201
  # @GPU(duration = 180)
202
  # def process_image(input_img):
203
  # # resize the input image
@@ -220,11 +220,13 @@ if __name__ == "__main__":
220
 
221
  # return input_img, input_img
222
 
223
- # demo = gr.Interface(
224
- # process_image,
225
- # gr.Image(),
226
- # [gr.Model3D(label="Input"), gr.Model3D(label="RaySt3R",)]
227
- # )
 
 
 
 
228
 
229
- # if __name__ == "__main__":
230
- # demo.launch()
 
9
  else:
10
  print("[WARNING] CUDA not available. Skipping warmup.")
11
 
12
+
 
 
13
  # import numpy as np
14
  # import gradio as gr
15
  # import rembg
 
196
  # rgb = output_np[:,:,:3]
197
  # return mask, rgb
198
 
199
+ def process_image(input_img):
200
+ return input_img, input_img
201
  # @GPU(duration = 180)
202
  # def process_image(input_img):
203
  # # resize the input image
 
220
 
221
  # return input_img, input_img
222
 
223
+ demo = gr.Interface(
224
+ process_image,
225
+ gr.Image(),
226
+ [gr.Model3D(label="Input"), gr.Model3D(label="RaySt3R",)]
227
+ )
228
+
229
+ if __name__ == "__main__":
230
+ dummy_warmup()
231
+ demo.launch()
232