Update src/streamlit_app.py

src/streamlit_app.py

@@ -11,12 +11,16 @@ os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
 
 # 📥 Download NLTK tokenizer
 nltk_data_path = "/tmp/nltk_data"
-nltk.download("punkt_tab", download_dir=nltk_data_path)
+nltk.download("punkt", download_dir=nltk_data_path)
 nltk.data.path.append(nltk_data_path)
 
 # 🔄 Load tokenizer and model from Hugging Face
-tokenizer = DistilBertTokenizerFast.from_pretrained("distilbert-base-uncased", cache_dir="/tmp/huggingface")
-model = TFDistilBertForSequenceClassification.from_pretrained("sundaram07/distilbert-sentence-classifier", cache_dir="/tmp/huggingface")
+tokenizer = DistilBertTokenizerFast.from_pretrained(
+    "distilbert-base-uncased", cache_dir="/tmp/huggingface"
+)
+model = TFDistilBertForSequenceClassification.from_pretrained(
+    "sundaram07/distilbert-sentence-classifier", cache_dir="/tmp/huggingface"
+)
 
 # 🔮 Predict AI probability for a sentence
 def predict_sentence_ai_probability(sentence):
@@ -49,18 +53,23 @@ st.set_page_config(page_title="AI Detector", layout="wide")
 st.title("🧠 AI Content Detector")
 st.markdown("This app detects the percentage of **AI-generated content** based on sentence-level analysis using DistilBERT.")
 
+# 📋 User Input Area
 user_input = st.text_area("📋 Paste your text below to check for AI-generated sentences:", height=300)
 
+# 🔘 Analyze Button
 if st.button("🔍 Analyze"):
     if not user_input.strip():
         st.warning("⚠️ Please enter some text to analyze.")
     else:
+        # Run analysis
         ai_percentage, analysis_results = predict_ai_generated_percentage(user_input)
 
+        # 🔍 Sentence-Level Results
         st.subheader("🔍 Sentence-level Analysis")
         for i, (sentence, prob, is_ai) in enumerate(analysis_results, start=1):
             label = "🟢 Human" if not is_ai else "🔴 AI"
             st.markdown(f"**{i}.** _{sentence}_\n\n → {label}")
 
+        # 📊 Final Result
         st.subheader("📊 Final Result")
         st.success(f"Estimated **AI-generated content**: **{ai_percentage:.2f}%**")