Update app_streamlit.py

app_streamlit.py

@@ -42,7 +42,7 @@ def caption_image(img):
 @st.cache_resource
 def load_toxic_classifier():
     # Load local fine-tuned model (full absolute path)
-    model_dir = "NightPrince/peft-distilbert-toxic-classifier/last-checkpoint"
+    model_dir = "NightPrince/peft-distilbert-toxic-classifier"
     base_model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=9)
     model = PeftModel.from_pretrained(base_model, model_dir)
     tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")