cybernatedArt commited on
Commit
f719dff
1 Parent(s): bce8e95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ labels = ['Acne / Rosacea',
20
  'Urticaria Hives', 'Vascular Tumors', 'Vasculitis', 'Warts Molluscum']
21
 
22
  def classify_image(photos):
23
- photos = photos.reshape((-1, 256, 256, 3))
24
  prediction = model.predict(photos).flatten()
25
  confidences = {labels[i]: float(prediction[i]) for i in range(23)}
26
  return confidences
@@ -53,7 +53,7 @@ gr.Interface(fn=classify_image,
53
  title = title,
54
  article = article,
55
  description = description,
56
- inputs=gr.inputs.Image(shape=(256, 256)),
57
  outputs=gr.outputs.Label(num_top_classes=4),
58
  examples=examples).launch()
59
 
 
20
  'Urticaria Hives', 'Vascular Tumors', 'Vasculitis', 'Warts Molluscum']
21
 
22
  def classify_image(photos):
23
+ photos = photos.reshape((-1, 224, 224, 3))
24
  prediction = model.predict(photos).flatten()
25
  confidences = {labels[i]: float(prediction[i]) for i in range(23)}
26
  return confidences
 
53
  title = title,
54
  article = article,
55
  description = description,
56
+ inputs=gr.inputs.Image(shape=(224, 224)),
57
  outputs=gr.outputs.Label(num_top_classes=4),
58
  examples=examples).launch()
59