Commit
·
368f089
1
Parent(s):
cf6ee13
fix cuda check
Browse files- enhanced_app.py +6 -3
enhanced_app.py
CHANGED
@@ -52,6 +52,9 @@ try:
|
|
52 |
except ImportError:
|
53 |
print("DepthAnythingV2 not available. Using precomputed examples only.")
|
54 |
DEPTH_AVAILABLE = False
|
|
|
|
|
|
|
55 |
|
56 |
# Set environment variable to disable xFormers
|
57 |
os.environ['XFORMERS_DISABLED'] = '1'
|
@@ -591,7 +594,7 @@ class PrecomputedExamplesManager:
|
|
591 |
"""Create a preview thumbnail from RGB input image."""
|
592 |
try:
|
593 |
image = Image.open(rgb_path)
|
594 |
-
image.thumbnail((
|
595 |
return image
|
596 |
|
597 |
except Exception as e:
|
@@ -898,7 +901,7 @@ def create_gradio_interface():
|
|
898 |
with gr.Row():
|
899 |
# Left Column: Controls and Input
|
900 |
with gr.Column(scale=1):
|
901 |
-
if
|
902 |
gr.Markdown("### Upload Image")
|
903 |
|
904 |
uploaded_image = gr.Image(
|
@@ -946,7 +949,7 @@ def create_gradio_interface():
|
|
946 |
gr.Markdown("#### 3D Point Cloud")
|
947 |
pointcloud_output = gr.Plot(label="Interactive 3D Point Cloud (Colored by Segmentation)")
|
948 |
|
949 |
-
if
|
950 |
upload_btn.click(
|
951 |
fn=process_uploaded_image,
|
952 |
inputs=[uploaded_image],
|
|
|
52 |
except ImportError:
|
53 |
print("DepthAnythingV2 not available. Using precomputed examples only.")
|
54 |
DEPTH_AVAILABLE = False
|
55 |
+
|
56 |
+
|
57 |
+
CUDA_AVAILABLE = torch.cuda.is_available()
|
58 |
|
59 |
# Set environment variable to disable xFormers
|
60 |
os.environ['XFORMERS_DISABLED'] = '1'
|
|
|
594 |
"""Create a preview thumbnail from RGB input image."""
|
595 |
try:
|
596 |
image = Image.open(rgb_path)
|
597 |
+
image.thumbnail((600, 450), Image.Resampling.LANCZOS)
|
598 |
return image
|
599 |
|
600 |
except Exception as e:
|
|
|
901 |
with gr.Row():
|
902 |
# Left Column: Controls and Input
|
903 |
with gr.Column(scale=1):
|
904 |
+
if CUDA_AVAILABLE:
|
905 |
gr.Markdown("### Upload Image")
|
906 |
|
907 |
uploaded_image = gr.Image(
|
|
|
949 |
gr.Markdown("#### 3D Point Cloud")
|
950 |
pointcloud_output = gr.Plot(label="Interactive 3D Point Cloud (Colored by Segmentation)")
|
951 |
|
952 |
+
if CUDA_AVAILABLE:
|
953 |
upload_btn.click(
|
954 |
fn=process_uploaded_image,
|
955 |
inputs=[uploaded_image],
|