- updates for webm generation

app.py

@@ -70,14 +70,15 @@ if generate_button:
         with st.spinner('Classifying the toxicity...'):
             time.sleep(2)  # Simulating delay
             model_output = [[{'label': 'LABEL_0', 'score': 0.9999998807907104},
-                             {'label': 'LABEL_1', 'score': 1.1919785395889282e-07}]]
+                             {'label': 'LABEL_1', 'score': 1.1919785395889282e-07},
+                             {'label': 'LABEL_2', 'score': 1.1919785395889282e-07}]]
             output = model_output[0]
 
             st.write("Toxicity Classifier Output:")
             for i in range(len(output)):
                 if output[i]['label'] == 'LABEL_0':
                     st.write(f"Non-Toxic Content: {output[i]['score'] * 100:.1f}%")
-                elif output[i]['label'] == 'LABEL_1':
+                elif output[i]['label'] == 'LABEL_2':
                     st.write(f"Toxic Content: {output[i]['score'] * 100:.1f}%")
                 else:
                     continue