LaoCzi's picture
Update app.py
e9189b8
raw
history blame
1.2 kB
import os
import openai
import requests
openai.api_key = os.getenv("OPEN_API_KEY")
import time
import gradio as gr
def check_openai_speed(api_key, prompt):
start_time = time.time()
response = openai.Completion.create(
model="text-davinci-003",
prompt="Q",
temperature=0.7,
max_tokens=1,
top_p=1.0,
frequency_penalty=0.0,
presence_penalty=0.0
)
end_time = time.time()
response_time = end_time - start_time
print(f'Response time: {response_time:.2f} seconds')
print(f'Response status code: {response.status_code}')
print(f'Response text: {response.text}')
response_time = "Time to answer: " + f'{end_time - start_time:.2f}' + " seconds"
status_code = "Response code: " + str(response.status_code)
return status_code, response_time
title = "Status API ChatGPT"
css="""
footer {visibility: hidden}
.gradio-container {padding-top: 100px}
"""
with gr.Blocks(css=css, title=title) as demo:
gr.HTML("<h1>Status API ChatGPT</h1>")
dt_1 = gr.outputs.HTML()
dt_2 = gr.outputs.HTML()
dt =[dt_1, dt_2]
demo.load(check_openai_speed, inputs=None, outputs=dt)
demo.launch(share=False, debug=True )