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