Edited OAI model params

app.py

@@ -23,7 +23,7 @@ class OpenAI_API:
     def __init__(self):
         self.openai_api_key = ''
         
-    def single_request(self, prompt_input):
+    def single_request(self, address_text):
         
         openai.api_type = "azure"
         openai.api_base = "https://damlaopenai.openai.azure.com/"
@@ -31,15 +31,19 @@ class OpenAI_API:
         openai.api_key = os.getenv("API_KEY")
 
         response = openai.Completion.create(
-            model="text-davinci-003",
-            prompt=prompt_input,
-            temperature=0,
-            max_tokens=300,
-            top_p=1,
-            frequency_penalty=0.0,
-            presence_penalty=0.0,
-            stop=["\n"],
-        )
+          engine="Davinci-003",
+          prompt=address_text,
+          temperature=0.,#9,
+          max_tokens=300,
+          top_p=1.0,
+          # n=1,
+          # logprobs=0,
+          # echo=False,
+          # stop=None,
+          frequency_penalty=0,
+          presence_penalty=0,
+          stop=["\n"],
+          best_of=1)
 
         return response
 
@@ -140,6 +144,18 @@ 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
 
 with gr.Blocks() as demo:
     gr.Markdown(