Spaces:
Runtime error
Runtime error
import gradio as gr | |
from gradio.inputs import Dropdown | |
import requests | |
""" | |
Use Gradio library to create a form with a textbox to store the input text. | |
And API_KEY textbox to store the key. | |
Use api.openai.com to get the response. | |
""" | |
def generate_text(input_text, api_key, temperature, max_tokens, frequency_penalty, presence_penalty, engine_id): | |
url = "https://api.openai.com/v1/engines/{engine_id}/completions".format(engine_id=engine_id) | |
data = { | |
"prompt": input_text, | |
"max_tokens": max_tokens, | |
"temperature": temperature, | |
"frequency_penalty": frequency_penalty, | |
"presence_penalty": presence_penalty | |
} | |
headers = {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"} | |
response = requests.post(url=url, json=data, headers=headers) | |
if response.status_code == 200: | |
return response.json()["choices"][0]["text"] | |
else: | |
return "Error: " + response.text | |
if __name__ == "__main__": | |
gr.Interface( | |
generate_text, # function to be called | |
[gr.inputs.Textbox(lines=1), gr.inputs.Textbox(lines=1), gr.inputs.Slider(minimum=0, maximum=1, step=0.01), | |
gr.inputs.Slider(minimum=32, maximum=500, step=1), gr.inputs.Slider(minimum=0, maximum=1, step=0.01), gr.inputs.Slider(minimum=0, maximum=1, step=0.01), gr.inputs.Dropdown(["davinci-codex", "cushman-codex"])], # input types | |
gr.outputs.Textbox() # output type |