Add HF token

app.py

@@ -28,6 +28,9 @@ import smtplib
 # if not os.path.isdir('images'):
 # os.mkdir('images')
 # print("create folder--->")
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "OCR-img-to-text-flag")
 print(os.getcwd())
 def get_device_ip_address():
 
@@ -113,22 +116,22 @@ def generate_ocr(Method,img):
         if Method == 'PaddleOCR':
             text_output = ocr_with_paddle(img)
         # save_details(Method,text_output,img)
-        sender="[email protected]"
-        password="httscgatatbbxxur"
-        reciever="[email protected]"
+        # sender="[email protected]"
+        # password="httscgatatbbxxur"
+        # reciever="[email protected]"
 
-        s = smtplib.SMTP('smtp.gmail.com', 587)
-        s.starttls()
-        s.ehlo()
-        s.login(sender,password)
+        # s = smtplib.SMTP('smtp.gmail.com', 587)
+        # s.starttls()
+        # s.ehlo()
+        # s.login(sender,password)
 
-        message = """Subject : Appointment Booking\n\n
-                Hello,
-        Your OCR generated successfully"""
-        s.sendmail(sender, reciever, message)
-        s.quit()
-        mailsend=1
-        print("Send mail successfully")
+        # message = """Subject : Appointment Booking\n\n
+        #         Hello,
+        # Your OCR generated successfully"""
+        # s.sendmail(sender, reciever, message)
+        # s.quit()
+        # mailsend=1
+        # print("Send mail successfully")
         return text_output
     
     except Exception as e:
@@ -197,6 +200,7 @@ demo = gr.Interface(
     description="Try OCR with different methods", 
     theme="darkpeach",
     css=".gradio-container {background-color: lightgray} #radio_div {background-color: #FFD8B4; font-size: 40px;}",
-    allow_flagging = "manual"
+    allow_flagging = "manual",
+    flagging_callback=hf_writer
 )
 demo.launch(enable_queue = False)