naveenvenkatesh commited on
Commit
ca08c30
1 Parent(s): dc0c610

Update Tags_Extractor.py

Browse files
Files changed (1) hide show
  1. Tags_Extractor.py +6 -1
Tags_Extractor.py CHANGED
@@ -31,7 +31,12 @@ class Tags:
31
  ]
32
 
33
  # Call OpenAI GPT-3.5-turbo
34
- chat_completion = self.client.chat.completions.create(model="ChatGPT", prompt=conversation, max_tokens=500)
 
 
 
 
 
35
  response = chat_completion.choices[0].message.content
36
  return response
37
 
 
31
  ]
32
 
33
  # Call OpenAI GPT-3.5-turbo
34
+ chat_completion = client.chat.completions.create(
35
+ model = "ChatGPT",
36
+ messages = conversation,
37
+ max_tokens=1000,
38
+ temperature=0
39
+ )
40
  response = chat_completion.choices[0].message.content
41
  return response
42