Update app.py

app.py

@@ -18,7 +18,7 @@ model = create_ViT()
 # Load saved weights
 model.load_state_dict(
     torch.load(
-        f="ViTHg.pth",
+        f="ViT.pth",
         map_location=torch.device("cpu"),
     )
 )
@@ -64,7 +64,7 @@ demo = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil"),
     outputs=[
-        gr.Label(num_top_classes=5, label="Predictions",type="key_value", num_format="{:.5f}"),
+        gr.Label(num_top_classes=5, label="Predictions"),
         gr.Number(label="Prediction time (s)"),
     ],
     examples=example_list,