Update app.py

app.py

@@ -112,8 +112,6 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
                         likeable=False,
                         layout='panel')
     
-    #chatbot = gr.Chatbot(label="Aira").style(height=500)
-    
     msg = gr.Textbox(label="Write a question or instruction to Aira ...", placeholder="What is the capital of Brazil?")
 
     # Parameters to control the generation
@@ -211,11 +209,13 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
                 
                 toxicity_threshold = 5
 
-        ordered_generations = sorted(zip(decoded_text, rewards, toxicities), key=lambda x: x[1], reverse=True)
-
         if safety == "On":
+            ordered_generations = sorted(zip(decoded_text, rewards, toxicities), key=lambda x: x[1], reverse=True)
             ordered_generations = [(x, y, z) for (x, y, z) in ordered_generations if z >= toxicity_threshold]
 
+        else:
+            ordered_generations = sorted(zip(decoded_text, rewards), key=lambda x: x[1], reverse=True)
+
         if len(ordered_generations) == 0:
           bot_message = """I apologize for the inconvenience, but it appears that no suitable responses meeting our safety standards could be identified. Unfortunately, this indicates that the generated content may contain elements of toxicity or may not help address your message. Your input is valuable to us, and we strive to ensure a safe and constructive conversation. Please feel free to provide further details or ask any other questions, and I will do my best to assist you."""