Update app.py

app.py

@@ -6,9 +6,9 @@ import pandas as pd
 import gradio as gr
 
 # ----------------------------------------------------
-# LOAD CAUSAL LM (GPT-J 6B = Best balance)
+# LOAD CAUSAL LM (Memory-safe model)
 # ----------------------------------------------------
-MODEL_NAME = "EleutherAI/gpt-j-6B"
+MODEL_NAME = "EleutherAI/gpt-neo-2.7B"
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -70,7 +70,6 @@ def detectgpt_score(sentence, perturbations=5):
     if not pert_scores:
         return 0
 
-    # DetectGPT signal
     return np.mean(pert_scores) - base
 
 
@@ -98,7 +97,6 @@ def classify_text(text):
         else:
             highlighted.append(f"<p style='color:green;font-weight:bold'>{s}</p>")
 
-    # Document-level score rescaled 0–100%
     avg = np.mean(scores)
     doc_ai = max(0, min(100, (avg + 1) * 50))
 
@@ -112,7 +110,7 @@ def classify_text(text):
 # GRADIO UI
 # ----------------------------------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("## 🧠 Writenix DetectGPT (GPT-J-6B Version)")
+    gr.Markdown("## 🧠 Writenix DetectGPT (GPT-Neo 2.7B Version)")
 
     text_input = gr.Textbox(
         label="Enter text",