Update app.py

app.py

@@ -2,14 +2,15 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 from datasets import load_dataset
 
-# Load your fine-tuned model from HF Hub
-MODEL = "yagnik12/AI_Text_Detecter_HanxiGuo_BiScope-Data"  # ✅ use fine-tuned model
+# ✅ Use your fine-tuned model (after running train.py)
+MODEL = "yagnik12/biscope-detector"
+
 tokenizer = AutoTokenizer.from_pretrained(MODEL)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL)
 
 detector = pipeline("text-classification", model=model, tokenizer=tokenizer, return_all_scores=True)
 
-# Load BiScope dataset (for demo examples)
+# Load some BiScope test examples for demo
 biscope = load_dataset("HanxiGuo/BiScope_Data", split="test[:20]")
 
 def detect_ai(text):
@@ -24,8 +25,8 @@ def detect_ai(text):
     }
 
 with gr.Blocks() as demo:
-    gr.Markdown("# AI vs Human Text Detector (BiScope)")
-    
+    gr.Markdown("# AI vs Human Text Detector (BiScope Dataset)")
+
     with gr.Row():
         inp = gr.Textbox(lines=5, placeholder="Enter text here...")
         out = gr.JSON()