sergiomar73 commited on
Commit
a8b1af3
·
1 Parent(s): 1326023

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -20,8 +20,7 @@ def classificate_with_gpt3(text, labels, engine="text-similarity-davinci-001"):
20
  frequency_penalty=0,
21
  presence_penalty=0
22
  )
23
- completion = response['data'][0]
24
- print(completion)
25
  return completion
26
 
27
  def gpt3_zero_shot_classification(text, labels):
 
20
  frequency_penalty=0,
21
  presence_penalty=0
22
  )
23
+ completion = response['choices'][0]['text'].strip()
 
24
  return completion
25
 
26
  def gpt3_zero_shot_classification(text, labels):