moved nerf function to utils

app.py

@@ -20,7 +20,7 @@ def get_text(input_img):
 # Submit button
 def get_parsed_address(input_img):
 
-    address_full_text = get_text(input_img)
+    address_full_text = utils.get_text(input_img)
     return openai_response(address_full_text)
 
 
@@ -76,18 +76,6 @@ def openai_response(ocr_input):
             resp[key] = ""
     return resp
 
-def ner_response(ocr_input):
-    API_URL = "https://api-inference.huggingface.co/models/deprem-ml/deprem-ner"
-    headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
-
-    def query(payload):
-        response = requests.post(API_URL, headers=headers, json=payload)
-        return response.json()
-
-    output = query({
-        "inputs": ocr_input,
-    })
-    return output
 
 # User Interface
 with gr.Blocks() as demo:

utils.py

@@ -3,6 +3,7 @@ import csv
 import json
 from deta import Deta
 import os
+import requests
 
 
 def preprocess_img(inp_image):
@@ -10,6 +11,7 @@ def preprocess_img(inp_image):
     gray_img = cv2.bitwise_not(gray)
     return gray_img
 
+
 def save_csv(mahalle, il, sokak, apartman):
     adres_full = [mahalle, il, sokak, apartman]
 
@@ -27,9 +29,25 @@ def get_json(mahalle, il, sokak, apartman):
 
 def write_db(data_dict):
     # 2) initialize with a project key
-    deta_key = os.getenv('DETA_KEY')
+    deta_key = os.getenv("DETA_KEY")
     deta = Deta(deta_key)
 
     # 3) create and use as many DBs as you want!
     users = deta.Base("deprem-ocr")
     users.insert(data_dict)
+
+
+def ner_response(ocr_input):
+    API_URL = "https://api-inference.huggingface.co/models/deprem-ml/deprem-ner"
+    headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
+
+    def query(payload):
+        response = requests.post(API_URL, headers=headers, json=payload)
+        return response.json()
+
+    output = query(
+        {
+            "inputs": ocr_input,
+        }
+    )
+    return output