deprem-ocr / openai_api.py
mertcobanov's picture
pep correction
f66b5f5
raw
history blame
738 Bytes
import openai
import os
class OpenAI_API:
def __init__(self):
self.openai_api_key = ""
def single_request(self, address_text):
openai.api_type = "azure"
openai.api_base = "https://damlaopenai.openai.azure.com/"
openai.api_version = "2022-12-01"
openai.api_key = os.getenv("API_KEY")
response = openai.Completion.create(
engine="Davinci-003",
prompt=address_text,
temperature=0.9,
max_tokens=256,
top_p=1.0,
n=1,
logprobs=0,
echo=False,
stop=None,
frequency_penalty=0,
presence_penalty=0,
best_of=1,
)
return response