New Output Field Support

app.py

@@ -553,7 +553,7 @@ def model_predict(word):
         draw.rectangle(bbox, fill=(200, 100, 0, 200))
         draw.text(pt1, classi, font=font, fill=(0,0,0,255))
     newWordImg = np.asarray(pil_image)
-    return newWordImg, output
+    return output, newWordImg
     '''
     output=''
     for i in range(0,len(final)):
@@ -571,8 +571,7 @@ def model_predict(word):
 
 import gradio as gr
 HF_TOKEN = os.getenv("HF_TOKEN", default ="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")
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "ocr-data")
-
-demo = gr.Interface(fn=model_predict, inputs= "paint", outputs=["image","text"],allow_flagging="auto",flagging_callback=hf_writer)
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "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")
 
+demo = gr.Interface(fn=model_predict, inputs= "paint", outputs=["text","image"],allow_flagging="auto",flagging_callback=hf_writer)
 demo.launch()