Update app.py

app.py

@@ -4,7 +4,7 @@ from toxic_detection import TextToxicDetector
 from toxic_detection import ImgToxicDetector
 
 text_model = TextToxicDetector()
-text_model.load('./text_detect_bert_16m.gz.tar')
+text_model.load('szzzzz/xlm-roberta-base-text-toxic')
 img_model = ImgToxicDetector()
 img_model.load('./toxic_detection_res50.gz.tar')
 
@@ -23,7 +23,7 @@ with gr.Blocks() as app:
         image_button_toxic = gr.Button("image_toxic")
     with gr.Tab("Toxic Text Detector"):
         text_input_toxic = gr.Textbox()
-        text_output_toxic = gr.Label(num_top_classes=6)
+        text_output_toxic = gr.Label(num_top_classes=1)
         text_button_toxic = gr.Button("text_toxic") 
     
     image_button_toxic.click(image_toxic_detect, inputs=image_input_toxic, outputs=image_output_toxic)