Update app.py
Browse files
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=
|
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
|