Update app.py

app.py

@@ -19,7 +19,11 @@ def respond(
     temperature,
     top_p,
 ):
-    res = answer(msg,model,tokenizer)
+    res = answer(message,model,tokenizer)
+    if res[1]>res[0]:
+        return "unsafe" # unsafe
+    else:
+        return "safe" # safe
     # messages = [{"role": "system", "content": system_message}]
 
     # for val in history:
@@ -43,7 +47,6 @@ def respond(
 
     #     response += token
     #     yield response
-    return res
 
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface