Update app.py

app.py

@@ -20,7 +20,7 @@ def gradio_wrapper(img):
     global model
     #print(np.shape(img))
     results = model.predict(img)  # predict on an image
-    print(results)
+    print(results['labels'].cpu().numpy().tolist())
     
     return cv2.putText(img, str(results[0]),(00, 185), cv2.FONT_HERSHEY_SIMPLEX, 1, 
                   (0, 0, 255),  2, cv2.LINE_AA, False)