Update app.py

app.py

@@ -2,8 +2,8 @@ import streamlit as st
 from transformers import DistilBertTokenizer, TFDistilBertForSequenceClassification
 import tensorflow as tf
 
-# Load the pre-trained model and tokenizer
-model_path = 'shukdevdatta123/Dreaddit_DistillBert_Stress_Model/'
+# Load the pre-trained model and tokenizer using the correct Hugging Face model repo ID
+model_path = 'shukdevdatta123/Dreaddit_DistillBert_Stress_Model'
 loaded_model = TFDistilBertForSequenceClassification.from_pretrained(model_path)
 loaded_tokenizer = DistilBertTokenizer.from_pretrained(model_path)