Update app.py

app.py

@@ -1,5 +1,6 @@
 import torch
 from transformers import ViTForImageClassification, ViTImageProcessor
+import torch.nn.functional as F
 from PIL import Image
 import gradio as gr
 
@@ -28,7 +29,7 @@ def predict(image):
 
 iface = gr.Interface(fn=predict, 
                      inputs=gr.Image(type="pil"), 
-                     outputs=[gr.Label(num_top_classes=1),gr.Label()]
+                     outputs=[gr.Label(num_top_classes=1, label="Class"), gr.Label(label="Score")],
                      title="Hateful Content Detection",
                      description="Upload an image to classify hateful gestures or symbols")