Update app.py

app.py

@@ -72,14 +72,17 @@ pipe = pipeline("image-classification", model=model, feature_extractor=processor
 
 def classify_image(input):
     face = grab_faces(np.array(input))
+    if face is not None:
+        x, y, w, h = face
+        cv.rectangle(input, (x, y), (x + w, y + h), (255, 255, 0), 2)
     face = Image.fromarray(face)
     result = pipe(face)
-    return result[0]["label"], result[0]["score"]
+    return result[0]["label"], result[0]["score"], input
 
 iface = gr.Interface(
     fn=classify_image,
     inputs="image",
-    outputs=["text", "number"],
+    outputs=["text", "number", "image"],
     title="Attraction Classifier - subjective",
     description="Takes in a (224, 224) image and outputs an attraction class: {\"pos\", \"neg\"}"
 )