Abedinho commited on
Commit
6449c65
1 Parent(s): d0a691e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ interface = gr.Interface(
65
  gr.Image(type="numpy", label="Upload an image of the exercise"),
66
  gr.Radio(['ResNet50', 'CNN'], label="Choose Model")
67
  ],
68
- outputs=gr.Label(num_top_classes=3, label="Prediction"),
69
  title="Exercise Classifier",
70
  description="Upload an image of an exercise and the classifier will predict the exercise.",
71
  css=custom_css # Apply the custom CSS
 
65
  gr.Image(type="numpy", label="Upload an image of the exercise"),
66
  gr.Radio(['ResNet50', 'CNN'], label="Choose Model")
67
  ],
68
+ outputs=gr.Label(num_top_classes=len(class_labels), label="Prediction"), # Show all classes
69
  title="Exercise Classifier",
70
  description="Upload an image of an exercise and the classifier will predict the exercise.",
71
  css=custom_css # Apply the custom CSS