Update app.py

app.py

@@ -70,11 +70,19 @@ processor = ViTImageProcessor.from_pretrained("ongkn/attraction-classifier")
 
 pipe = pipeline("image-classification", model=model, feature_extractor=processor)
 
-def classify_image(inp):
-    face = grab_faces(np.array(inp))
+def classify_image(input):
+    face = grab_faces(np.array(input))
     face = Image.fromarray(face)
     result = pipe(face)
     return result[0]["label"], result[0]["score"]
 
-iface = gr.Interface(fn=classify_image, inputs="image", outputs=["text", "number"])
-iface.launch()
+iface = gr.Interface(
+    fn=classify_image,
+    inputs="image",
+    outputs=[
+        {"type": "text", "label": "attraction class"},
+        {"type": "number", "label": "score (confidence)"}
+    ],
+    description="Takes in a (224, 224) image and outputs an attraction class: {\"pos\", \"neg\"}"
+)
+iface.launch()