Update app.py

app.py

@@ -204,8 +204,8 @@ def image_classifier(inp):
 
     # If input not valid, return dummy data or raise error
     if inp is None:
-        return {'cat': 0.3, 'dog': 0.7}
-
+        if inp is None:
+            return {'Actinic Keratosis': 0.0, 'Basal Cell Carcinoma': 0.0, 'Benign Keratosis': 0.0, 'Dermatofibroma': 0.0, 'Melanoma': 0.0, 'Melanocytic Nevi': 0.0, 'Vascular Lesion': 0.0}
     # preprocess
     image = preprocess_image(inp)
     image = image.to(dtype=torch.float32)