Update app.py
Browse files
app.py
CHANGED
@@ -9,9 +9,7 @@ def dummy_warmup():
|
|
9 |
else:
|
10 |
print("[WARNING] CUDA not available. Skipping warmup.")
|
11 |
|
12 |
-
|
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 |
-
|
224 |
-
|
225 |
-
|
226 |
-
|
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 |
|
|
|
|