Update app.py

app.py

@@ -20,9 +20,15 @@ def gradio_wrapper(img):
     global model
     #print(np.shape(img))
     results = model.predict(img)  # predict on an image
-    print(results[0].boxes.cls)
-    
-    return cv2.putText(img, str(results[0].boxes.cls),(00, 185), cv2.FONT_HERSHEY_SIMPLEX, 1, 
+    try:
+        if max(results[0].boxes.cls) == 0:
+            text = "Man"
+        if max(results[0].boxes.cls) == 1:
+            text = "Women"
+    except:
+        pass
+        
+    return cv2.putText(img, text,(00, 185), cv2.FONT_HERSHEY_SIMPLEX, 1, 
                   (0, 0, 255),  2, cv2.LINE_AA, False)
     # return results