Update app.py

app.py

@@ -21,7 +21,7 @@ def gradio_wrapper(img):
     #print(np.shape(img))
     results = model.predict(img)  # predict on an image
     
-    return cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    return cv2.putText(img, str(results[0]), cv2.LINE_AA, False)
 demo = gr.Interface(
     gradio_wrapper,
     #gr.Image(source="webcam", streaming=True, flip=True),