Spaces:

Aumkeshchy2003
/

ViT_For_100_Class

Sleeping

Aumkeshchy2003 commited on Nov 4, 2025

Commit

7028c06

verified ·

1 Parent(s): cafcdec

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -220,7 +220,7 @@ def predict(img: Image.Image):
 iface = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil"),
-    outputs=gr.Label(num_top_classes=1),
     title="Hybrid ViT+CNN CIFAR-100 Classifier",
     description="Upload a 32x32 image, and the model predicts the CIFAR-100 class.",
     examples=["Feng-shui-fish-acquarium_0_1200.jpg.webp",

 iface = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil"),
+    outputs=gr.Label(num_top_classes=5, label="Top-5 Predictions"),
     title="Hybrid ViT+CNN CIFAR-100 Classifier",
     description="Upload a 32x32 image, and the model predicts the CIFAR-100 class.",
     examples=["Feng-shui-fish-acquarium_0_1200.jpg.webp",