Update app.py

app.py

@@ -8,20 +8,17 @@ import gradio as gr
 
 
 def check_openai_speed(api_key, prompt):
-    headers = {
-        'Content-Type': 'application/json',
-        'Authorization': f'Bearer {openai.api_key}',
-    }
-    data = {
-        'prompt': "q",
-        'temperature': 0.5,
-        'max_tokens': 1,
-        'top_p': 1,
-        'frequency_penalty': 0,
-        'presence_penalty': 0
-    }
+
     start_time = time.time()
-    response = requests.post('https://api.openai.com/v1/engines/davinci-codex/completions', headers=headers, json=data)
+    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')