Upload app.py

app.py

@@ -1,16 +1,28 @@
 import gradio as gr
-from authormist import Humanizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-humanizer = Humanizer()
+# Load the AuthorMist model and tokenizer
+model_name = "authormist/authormist-originality"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
-def humanize(text):
-    return humanizer.humanize(text)
+# Function to humanize input text
+def humanize_text(text):
+    prompt = f"Please paraphrase the following text to make it human-like:\n\n{text}\n\nParaphrased text:"
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(inputs.input_ids, max_new_tokens=512, temperature=0.7, top_p=0.9)
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-demo = gr.Interface(
-    fn=humanize,
-    inputs="textbox",
-    outputs="textbox",
-    title="AuthorMist AI Humanizer"
-)
+    # Extract only the paraphrased part
+    if "Paraphrased text:" in response:
+        return response.split("Paraphrased text:")[1].strip()
+    return response.strip()
 
-demo.launch()
+# Launch Gradio UI
+gr.Interface(
+    fn=humanize_text,
+    inputs=gr.Textbox(lines=10, placeholder="Paste AI-written text here..."),
+    outputs="text",
+    title="AuthorMist AI Humanizer",
+    description="Turns AI-generated text into human-like writing to reduce detection by tools like GPTZero."
+).launch()