research14 commited on
Commit
b2e68d1
·
1 Parent(s): e98cf95
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import time
@@ -15,9 +17,9 @@ template_single = '''Please output any <{}> in the following sentence one per li
15
  os.environ['OPENAI_API_TOKEN'] = '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'
16
  openai.api_key = os.environ['OPENAI_API_TOKEN']
17
 
18
- def chat(system_prompt, user_prompt, model = 'gpt3.5', temperature = 0, verbose = False):
19
  ''' Normal call of OpenAI API '''
20
- response = openai.Completion.create(
21
  temperature = temperature,
22
  model=model,
23
  messages=[
 
1
+ pip install openai==0.28
2
+
3
  import gradio as gr
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
  import time
 
17
  os.environ['OPENAI_API_TOKEN'] = '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'
18
  openai.api_key = os.environ['OPENAI_API_TOKEN']
19
 
20
+ def chat(system_prompt, user_prompt, model = 'gpt-3.5-turbo', temperature = 0, verbose = False):
21
  ''' Normal call of OpenAI API '''
22
+ response = openai.ChatCompletion.create(
23
  temperature = temperature,
24
  model=model,
25
  messages=[