bug fix

app.py

@@ -55,7 +55,7 @@ if st.button("Analyze"):
     else:
         with st.spinner("Analyzing toxicity..."):
             if selected_model == "Olivernyu/finetuned_bert_base_uncased":
-                toxicity_detector = load_finetune_model(selected_model)
+                toxicity_detector = load_model(selected_model)
                 outputs = toxicity_detector(text, top_k=2)
                 category_names = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
                 results = []