Update app.py

app.py

@@ -16,17 +16,13 @@ def preprocess_image(image):
     return image
 
 def predict_image(model, image):
-    
-    if torch.sum(image) == 0:
-        return 404  
-    
     model.eval()
     with torch.no_grad():
         output = model(image)
-
+    print(output)
     _, predicted = torch.max(output.data, 1)
     return predicted.item()
-    
+
 def main():
    
     model = SimpleCNN()