Spaces:
Runtime error
Runtime error
File size: 1,268 Bytes
5bdee70 303e229 5bdee70 303e229 5bdee70 dd9d76c 5bdee70 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import openai
poem = """Write what is asked of you:
---
{input}
---
This is the result: """
def set_openai_key(key):
"""Sets OpenAI key."""
openai.api_key = key
class GeneralModel:
def __init__(self):
print("Model Intilization--->")
# set_openai_key(API_KEY)
def query(self, prompt, myKwargs={}):
"""
wrapper for the API to save the prompt and the result
"""
# arguments to send the API
kwargs = {
"engine": "text-davinci-003",
"temperature": 0.85,
"max_tokens": 2400,
"best_of": 1,
"top_p": 1,
"frequency_penalty": 0.5,
"presence_penalty": 0.5,
"stop": ["###"],
}
for kwarg in myKwargs:
kwargs[kwarg] = myKwargs[kwarg]
r = openai.Completion.create(prompt=prompt, **kwargs)["choices"][0][
"text"
].strip()
return r
def model_prediction(self, input, api_key):
"""
wrapper for the API to save the prompt and the result
"""
# Setting the OpenAI API key got from the OpenAI dashboard
set_openai_key(api_key)
output = self.query(poem.format(input = input))
return output |