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 |
|
|
|
|
|
|