ganteng88 commited on
Commit
5f4e838
·
1 Parent(s): 4a047be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -205,7 +205,8 @@ def image_classifier(inp):
205
  # If input not valid, return dummy data or raise error
206
  if inp is None:
207
  if inp is None:
208
- 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}
 
209
  # preprocess
210
  image = preprocess_image(inp)
211
  image = image.to(dtype=torch.float32)
 
205
  # If input not valid, return dummy data or raise error
206
  if inp is None:
207
  if inp is None:
208
+ return {'cat': 0.3, 'dog': 0.7}
209
+ #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}
210
  # preprocess
211
  image = preprocess_image(inp)
212
  image = image.to(dtype=torch.float32)