Update app.py

app.py

@@ -1,16 +1,20 @@
 import gradio
-import cv2
-import os
-from ktp_reader import process_image
-import logging
-logging.getLogger().setLevel(logging.ERROR)
+import Levenshtein
+
+def calculate_distance(org_text, ocr_str):
+    # split words and characters
+    ocr_words = ocr_str.split()
+    org_words = org_str.split()
+    ocr_chars = list(ocr_str)
+    org_chars = list(org_str)
+    cer = Levenshtein.distance(tran_chars, ocr_chars) / len(tran_chars)
+    return return f"The Character Error Rate (CER): {cer}"
 
 iface = gradio.Interface(
-    fn=process_image,
-    inputs='image',
+    fn=calculate_distance,
+    inputs=['text','text'],
     outputs='text',
-    title='Indonesian ID Data Extraction', 
-    description="This space includes a solution to extract Indonesian ID data using Tesseract, PaddleOCR, OpenCV, and TheFuzz.",
-    examples=["id.jpg"])  
+    title='OCR Character Error Rate (CER) Calculation Using Levenshtein Distance ' 
+    description="In this space, you can enter the OCR text result and the original text to calculate ")  
     
 iface.launch()