tanveeshsingh commited on
Commit
b4e15d7
·
1 Parent(s): 04f84ab
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -60,6 +60,7 @@ Now, please output the following as a JSON object:
60
  max_tokens=400
61
  )
62
  output = completion.choices[0].message.content
 
63
  if output.output==1:
64
  return 'safe'
65
  else:
 
60
  max_tokens=400
61
  )
62
  output = completion.choices[0].message.content
63
+ output = json.loads(output)
64
  if output.output==1:
65
  return 'safe'
66
  else: