Curranj commited on
Commit
9d3cf82
·
verified ·
1 Parent(s): ff66bb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -2,20 +2,19 @@ import openai
2
  import gradio as gr
3
  import os
4
 
5
- # Function to get GPT response
6
  def gpt_response(prompt):
7
  openai.api_key = os.getenv("OPENAI_API_KEY") # Ensure your API key is set as an environment variable
8
- response = openai.ChatCompletion.create(
9
- model="gpt-4", # Specify the GPT model to use
10
- messages=[{"role": "system", "content": "You are a helpful assistant."},
11
- {"role": "user", "content": prompt}],
12
  temperature=0,
13
  max_tokens=750,
14
  top_p=1,
15
  frequency_penalty=0.0,
16
  presence_penalty=0.0
17
  )
18
- return response['choices'][0]['message']['content']
19
 
20
  # Gradio interface function
21
  def greet(prompt):
 
2
  import gradio as gr
3
  import os
4
 
5
+ # Function to get GPT response using text completions
6
  def gpt_response(prompt):
7
  openai.api_key = os.getenv("OPENAI_API_KEY") # Ensure your API key is set as an environment variable
8
+ response = openai.Completion.create(
9
+ model="text-davinci-003", # Specify a non-chat model
10
+ prompt=prompt,
 
11
  temperature=0,
12
  max_tokens=750,
13
  top_p=1,
14
  frequency_penalty=0.0,
15
  presence_penalty=0.0
16
  )
17
+ return response['choices'][0]['text'].strip()
18
 
19
  # Gradio interface function
20
  def greet(prompt):