error handler
Browse files
app.py
CHANGED
|
@@ -61,6 +61,8 @@ def loading_data(img):
|
|
| 61 |
|
| 62 |
|
| 63 |
def predict(img):
|
|
|
|
|
|
|
| 64 |
"""the main process of inference"""
|
| 65 |
test_loader = loading_data(img)
|
| 66 |
#model = SASNet()
|
|
|
|
| 61 |
|
| 62 |
|
| 63 |
def predict(img):
|
| 64 |
+
if img is None:
|
| 65 |
+
return "No image selected", plt.figure(np.ones((30,30)))
|
| 66 |
"""the main process of inference"""
|
| 67 |
test_loader = loading_data(img)
|
| 68 |
#model = SASNet()
|