Update app.py

app.py

@@ -1,7 +1,7 @@
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch, gradio as gr
 
-
+# Load Model
 model_name = "alykassem/FLAN-T5-Paraphraser"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
@@ -10,28 +10,36 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)
 model.eval()
 
-def paraphrase_dipper(text, diversity=0.5, reordering=0.5):
-    inputs = tokenizer([text], return_tensors="pt", truncation=True, padding=True).to(device)
+# Paraphrasing Function
+def paraphrase_flan(text, diversity=0.7, temperature=0.9):
+    if not text.strip():
+        return "⚠️ Please enter some text"
+
+    inputs = tokenizer([f"paraphrase: {text}"], return_tensors="pt", truncation=True, padding=True).to(device)
+    
     outputs = model.generate(
         **inputs,
         max_new_tokens=256,
-        top_p=1.0,
-        diversity_penalty=float(diversity),
-        num_reorder=float(reordering),
         do_sample=True,
+        top_p=0.92,  # nucleus sampling
+        temperature=float(temperature),  # creativity
+        diversity_penalty=float(diversity),  # encourages lexical variety
+        num_return_sequences=1
     )
+
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
+# Gradio UI
 iface = gr.Interface(
-    fn=paraphrase_dipper,
+    fn=paraphrase_flan,
     inputs=[
         gr.Textbox(lines=8, placeholder="Paste full text here..."),
-        gr.Slider(0.0, 1.0, step=0.1, value=0.5, label="Lexical Diversity"),
-        gr.Slider(0.0, 1.0, step=0.1, value=0.5, label="Reordering Amount")
+        gr.Slider(0.0, 1.0, step=0.1, value=0.7, label="Lexical Diversity"),
+        gr.Slider(0.5, 1.5, step=0.1, value=0.9, label="Temperature")
     ],
     outputs=gr.Textbox(label="Paraphrased & Humanized Text"),
-    title="DIPPER Paraphraser (AI-Detector Evading)",
-    description="Paraphrase full text with diversity and reordering control to reduce AI detection."
+    title="FLAN-T5 Paraphraser (Humanizer)",
+    description="High-quality paraphrasing with adjustable diversity and temperature to reduce AI detection."
 )
 
 iface.launch()