Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ def preprocess_image(image):
|
|
18 |
def predict_image(model, image):
|
19 |
|
20 |
if torch.sum(image) == 0:
|
21 |
-
return
|
22 |
|
23 |
model.eval()
|
24 |
with torch.no_grad():
|
|
|
18 |
def predict_image(model, image):
|
19 |
|
20 |
if torch.sum(image) == 0:
|
21 |
+
return 404
|
22 |
|
23 |
model.eval()
|
24 |
with torch.no_grad():
|