Update app.py

app.py

@@ -122,7 +122,7 @@ Generate OCR
 def generate_ocr(Method,img):
     try:
         text_output = ''
-
+        add_csv = []
         print("Method___________________",Method)
         if Method == 'EasyOCR':
             text_output = ocr_with_easy(img)
@@ -135,11 +135,10 @@ def generate_ocr(Method,img):
         print("^^^^^^^^^^^^^^^",new_data)
         imge = Image.fromarray(new_data.astype(np.uint8),'RGB')
         # imge.show()
+        add_csv = [Method,imge,text_output]
         with open(DATA_FILE, "a") as csvfile:
-            writer = csv.DictWriter(csvfile, fieldnames=["method", "image", "generated_text"])
-            writer.writerow(
-                {"method": Method, "image": imge, "generated_text": text_output}
-            )
+            writer = csv.Writer(csvfile)
+            writer.writerow(add_csv)
         commit_url = repo.push_to_hub()
         print(commit_url)
         # save_details(Method,text_output,img)