Spaces:

yuragoithf
/

mlg_image_classification

Sleeping

yuragoithf commited on Jun 3, 2023

Commit

33007ff

1 Parent(s): 7c5859b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ def classify_image(image):
 inputs = gr.inputs.Image(type="pil", label="Upload an image")
 # outputs = gr.outputs.HTML() #uncomment for single class output
-outputs = gr.outputs.Label(num_top_classes=5)
 title = "<h1 style='text-align: center;'>Image Classifier</h1>"
 description = "Upload an image and get the predicted class."

 inputs = gr.inputs.Image(type="pil", label="Upload an image")
 # outputs = gr.outputs.HTML() #uncomment for single class output
+outputs = gr.outputs.Label(num_top_classes=4)
 title = "<h1 style='text-align: center;'>Image Classifier</h1>"
 description = "Upload an image and get the predicted class."