Upload 2 files

app.py

@@ -0,0 +1,49 @@
+
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+import gradio as gr
+
+model_name = "ramsrigouthamg/t5_paraphraser"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+paraphraser = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
+
+def rewrite_text(text, style):
+    if not text.strip():
+        return "Please enter some text."
+
+    if style == "Gen Z":
+        prompt = f"Rewrite the following text in a funny Gen Z tone with slang, emojis, and internet expressions:\n\n{text}"
+    elif style == "Poetic":
+        prompt = f"Rewrite the following text in a poetic and artistic style:\n\n{text}"
+    elif style == "Formal":
+        prompt = f"Rewrite the following text in a formal, professional tone:\n\n{text}"
+    elif style == "Friendly":
+        prompt = f"Rewrite the following text in a friendly and conversational style:\n\n{text}"
+    else:
+        prompt = f"Rewrite the following text:\n\n{text}"
+
+    response = paraphraser(prompt, max_length=100)[0]['generated_text']
+    return response.replace(prompt, "").strip()
+
+with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
+    gr.Markdown(
+        """
+        <h1 style="text-align: center;">📝 Rewrite My Text</h1>
+        <p style="text-align: center;">Transform your text into <b>fun, formal, poetic, or Gen Z</b> styles using AI! 🚀</p>
+        """,
+    )
+
+    with gr.Row():
+        with gr.Column():
+            input_text = gr.Textbox(label="Input Text", placeholder="Enter your sentence here...", lines=4)
+            style = gr.Dropdown(label="Choose Style", choices=["Gen Z", "Formal", "Poetic", "Friendly"], value="Gen Z")
+            submit_button = gr.Button("✨ Submit")
+            clear_button = gr.Button("🧹 Clear")
+
+        with gr.Column():
+            output_text = gr.Textbox(label="Rewritten Text", placeholder="Your rewritten sentence will appear here...", lines=4)
+
+    submit_button.click(fn=rewrite_text, inputs=[input_text, style], outputs=output_text)
+    clear_button.click(fn=lambda: ("", ""), inputs=[], outputs=[input_text, output_text])
+
+demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+
+transformers
+gradio
+torch
+sentencepiece