Update app.py

app.py

@@ -19,30 +19,30 @@ import socket
 from huggingface_hub import HfApi
 import smtplib
 
-# HF_TOKEN = os.getenv('HF_TOKEN')
-# print("$$$$$$$$$$$$$$$$$$$$$$$$$$$$",type(HF_TOKEN))
-# hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN,'OCR-image-to-text',True,True)
-
-
-DATASET_REPO_URL = "https://huggingface.co/datasets/pragnakalp/OCR-img-to-text"
-DATA_FILENAME = "ocr_data.csv"
-DATA_FILE = os.path.join("ocr_data", DATA_FILENAME)
-HF_TOKEN = os.environ.get("HF_TOKEN")
-DATASET_REPO_ID = "pragnakalp/OCR-img-to-text"
-print("is none?", HF_TOKEN is None)
-try:
-    hf_hub_download(
-        repo_id=DATASET_REPO_ID,
-        filename=DATA_FILENAME,
-        cache_dir=DATA_DIRNAME,
-        force_filename=DATA_FILENAME
-    )
-except:
-    print("file not found")
-
-repo = Repository(
-    local_dir="ocr_data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
-)
+HF_TOKEN = os.getenv('HF_TOKEN')
+print("$$$$$$$$$$$$$$$$$$$$$$$$$$$$",type(HF_TOKEN))
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN,'OCR-img-to-text')
+
+
+# # DATASET_REPO_URL = "https://huggingface.co/datasets/pragnakalp/OCR-img-to-text"
+# DATA_FILENAME = "ocr_data.csv"
+# DATA_FILE = os.path.join("ocr_data", DATA_FILENAME)
+# HF_TOKEN = os.environ.get("HF_TOKEN")
+# DATASET_REPO_ID = "pragnakalp/OCR-img-to-text"
+# print("is none?", HF_TOKEN is None)
+# try:
+#     hf_hub_download(
+#         repo_id=DATASET_REPO_ID,
+#         filename=DATA_FILENAME,
+#         cache_dir=DATA_DIRNAME,
+#         force_filename=DATA_FILENAME
+#     )
+# except:
+#     print("file not found")
+
+# repo = Repository(
+#     local_dir="ocr_data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+# )
 
 def get_device_ip_address():
 
@@ -127,13 +127,13 @@ def generate_ocr(Method,img):
         if Method == 'PaddleOCR':
             text_output = ocr_with_paddle(img)
 
-        with open(DATA_FILE, "a") as csvfile:
-            writer = csv.DictWriter(csvfile, fieldnames=["method", "image", "generated_text"])
-            writer.writerow(
-                {"method": Method, "image": img, "generated_text": text_output}
-            )
-        commit_url = repo.push_to_hub()
-        print(commit_url)
+        # with open(DATA_FILE, "a") as csvfile:
+        #     writer = csv.DictWriter(csvfile, fieldnames=["method", "image", "generated_text"])
+        #     writer.writerow(
+        #         {"method": Method, "image": img, "generated_text": text_output}
+        #     )
+        # commit_url = repo.push_to_hub()
+        # print(commit_url)
         
         # save_details(Method,text_output,img)
         # sender="[email protected]"
@@ -220,7 +220,7 @@ demo = gr.Interface(
     description="Try OCR with different methods",
     css=".gradio-container {background-color: lightgray} #radio_div {background-color: #FFD8B4; font-size: 40px;}",
     allow_flagging = "manual",
- #    flagging_dir = "flagged",
-	# flagging_callback=hf_writer
+    flagging_dir = "flagged",
+	flagging_callback=hf_writer
 )
 demo.launch(enable_queue = False)