Spaces:

collinear-ai
/

collinear-guard-demo

Sleeping

tanveeshsingh commited on Sep 18, 2024

Commit

e7c37c2

1 Parent(s): 6807272

Quick fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ Now, please output the following as a JSON object:
     output = completion.choices[0].message.content
     output = json.loads(output)
     print(output)
-    if output.output==1:
         return 'safe'
     else:
         return 'unsafe'

     output = completion.choices[0].message.content
     output = json.loads(output)
     print(output)
+    if output['output']==1:
         return 'safe'
     else:
         return 'unsafe'