TRaw commited on
Commit
1396b24
·
1 Parent(s): 8c6bb6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -9
app.py CHANGED
@@ -15,15 +15,26 @@ prompt = "The following is a conversation with an AI assistant. The assistant is
15
 
16
  def openai_create(prompt):
17
 
18
- response = openai.Completion.create(
19
- model="text-davinci-003",
20
- prompt=prompt,
21
- temperature=0.4,
22
- max_tokens=12500,
23
- top_p=1,
24
- frequency_penalty=0,
25
- presence_penalty=0.6,
26
- stop=[" Human:", " AI:"]
 
 
 
 
 
 
 
 
 
 
 
27
  )
28
 
29
  return response.choices[0].text
 
15
 
16
  def openai_create(prompt):
17
 
18
+ openai.api_key = os.getenv("OPENAI_API_KEY")
19
+
20
+ response = openai.ChatCompletion.create(
21
+ model="gpt-3.5-turbo-16k",
22
+ messages=[
23
+ {
24
+ "role": "system",
25
+ "content": "You are a helpful assistant!"
26
+ },
27
+ {
28
+ "role": "user",
29
+ "content": ""
30
+ }
31
+ ],
32
+ temperature=0.4,
33
+ max_tokens=12500,
34
+ top_p=1,
35
+ frequency_penalty=0,
36
+ presence_penalty=0
37
+ stop=[" Human:", " AI:"]
38
  )
39
 
40
  return response.choices[0].text