Update app.py

app.py

@@ -72,12 +72,11 @@ 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)
+    if face is None:
+        return "No face detected", 0, input
     face = Image.fromarray(face)
     result = pipe(face)
-    return result[0]["label"], result[0]["score"], input
+    return result[0]["label"], result[0]["score"], face
 
 iface = gr.Interface(
     fn=classify_image,