naveenvenkatesh commited on
Commit
10bbb62
1 Parent(s): 4683caa

Update headings_extractor.py

Browse files
Files changed (1) hide show
  1. headings_extractor.py +10 -6
headings_extractor.py CHANGED
@@ -45,12 +45,16 @@ class HeadingsExtractor:
45
  ]
46
 
47
  # Call OpenAI GPT-3.5-turbo
48
- chat_completion =self.client.chat.completions.create(
49
- model = "gpt-3.5-turbo",
50
- messages = conversation,
51
- max_tokens=1000,
52
- temperature=0
53
- )
 
 
 
 
54
  response = chat_completion.choices[0].message.content
55
  return response
56
 
 
45
  ]
46
 
47
  # Call OpenAI GPT-3.5-turbo
48
+ chat_completion =openai.ChatCompletion.create(
49
+ engine="ChatGPT",
50
+ messages = conversation,
51
+ temperature=0.7,
52
+ max_tokens=800,
53
+ top_p=0.95,
54
+ frequency_penalty=0,
55
+ presence_penalty=0,
56
+ stop=None
57
+ )
58
  response = chat_completion.choices[0].message.content
59
  return response
60