Spaces:
Running
Running
Handle edge case better
Browse files- demo/src/gui.py +1 -1
demo/src/gui.py
CHANGED
@@ -105,7 +105,7 @@ class WebUI:
|
|
105 |
return "./prediction.obj"
|
106 |
|
107 |
def download_prediction(self):
|
108 |
-
if (
|
109 |
LOGGER.error(
|
110 |
"The prediction is not available or ready to download. Wait until the result is available in the 3D viewer."
|
111 |
)
|
|
|
105 |
return "./prediction.obj"
|
106 |
|
107 |
def download_prediction(self):
|
108 |
+
if (self.filename is None) or (self.extension is None):
|
109 |
LOGGER.error(
|
110 |
"The prediction is not available or ready to download. Wait until the result is available in the 3D viewer."
|
111 |
)
|