Vermeer commited on
Commit
35dd703
1 Parent(s): b657d51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def classify(img):
12
  return {categories[i]: float(pred[i]) for i in range(3)}
13
 
14
  image = gr.inputs.Image(shape=(224, 224))
15
- label = gr.outputs.Label(num_top_classes=2)
16
  examples = ["Normal.png", "Tuberculosis.png", "Pneumonia.jpeg"]
17
 
18
 
 
12
  return {categories[i]: float(pred[i]) for i in range(3)}
13
 
14
  image = gr.inputs.Image(shape=(224, 224))
15
+ label = gr.outputs.Label(num_top_classes=3)
16
  examples = ["Normal.png", "Tuberculosis.png", "Pneumonia.jpeg"]
17
 
18