Curranj commited on
Commit
8746a42
·
verified ·
1 Parent(s): f7af7f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -11
app.py CHANGED
@@ -2,25 +2,24 @@ import openai
2
  import gradio as gr
3
  import os
4
 
5
- def gpt3(texts):
6
- client = openai.OpenAI(api_key=os.environ["Secret"]) # Create client instance
7
- response = client.chat.completions.create( # Use chat completions
8
- model="gpt-4-turbo-preview", # Updated model name
9
- messages=[{"role": "user", "content": texts}], # New messages format
 
10
  temperature=0,
11
  max_tokens=750,
12
  top_p=1,
13
  frequency_penalty=0.0,
14
- presence_penalty=0.0,
15
- stop=[";", "/*", "</code>"]
16
  )
17
- x = response.choices[0].message.content # Updated response structure
18
- return x
19
 
20
  def greet(prompt):
21
- txt= f'''/*Prompt: {prompt}*/ \n'''
22
  sql = gpt3(txt)
23
  return sql
24
 
25
- iface = gr.Interface(greet, inputs = ["text"])
26
  iface.launch()
 
2
  import gradio as gr
3
  import os
4
 
5
+ def gpt3(prompt):
6
+ openai.api_key = os.getenv("OPENAI_API_KEY")
7
+ response = openai.ChatCompletion.create(
8
+ model="gpt-4",
9
+ messages=[{"role": "system", "content": "You are a helpful assistant."},
10
+ {"role": "user", "content": 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]['message']['content']
 
18
 
19
  def greet(prompt):
20
+ txt = f"/*Prompt: {prompt}*/"
21
  sql = gpt3(txt)
22
  return sql
23
 
24
+ iface = gr.Interface(fn=greet, inputs="text", outputs="text")
25
  iface.launch()