ptdevtest

Running

App Files Files Community

DCComplex commited on 20 days ago

Commit

e7739fb

1 Parent(s): 2dbfc5b

Add application file

Browse files

Files changed (4) hide show

Dockerfile +16 -0
app.py +74 -0
requirements.txt +5 -0
templates/index.html +89 -0

Dockerfile ADDED Viewed

	@@ -0,0 +1,16 @@

+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+FROM python:3.9
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+WORKDIR /app
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py ADDED Viewed

	@@ -0,0 +1,74 @@

+# app.py
+from flask import Flask, render_template, request, jsonify
+from transformers import AutoTokenizer, AutoModelForSeq2SeqGeneration
+import torch
+app = Flask(__name__)
+# Load the model and tokenizer
+model_name = "t5-base"  # T5 is better suited for paraphrasing
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqGeneration.from_pretrained(model_name)
+def paraphrase_text(text, num_variants=3, max_length=100):
+    """
+    Paraphrase the input text using T5 model
+    """
+    if not text.strip():
+        return []
+    # Prepare the input text
+    prefix = "paraphrase: "
+    input_text = prefix + text
+    input_ids = tokenizer.encode(input_text, return_tensors="pt", max_length=max_length, truncation=True)
+    # Generate paraphrases
+    outputs = model.generate(
+        input_ids,
+        num_return_sequences=num_variants,
+        num_beams=num_variants * 2,
+        max_length=max_length,
+        temperature=0.7,
+        top_k=50,
+        top_p=0.95,
+        do_sample=True,
+        repetition_penalty=1.2,
+        early_stopping=True
+    )
+    # Decode and return the generated paraphrases
+    paraphrases = []
+    for output in outputs:
+        paraphrased = tokenizer.decode(output, skip_special_tokens=True)
+        if paraphrased.lower() != text.lower():  # Avoid exact duplicates
+            paraphrases.append(paraphrased)
+    return paraphrases
+@app.route("/", methods=["GET", "POST"])
+def index():
+    paraphrased_texts = []
+    original_text = ""
+    error_message = ""
+    if request.method == "POST":
+        original_text = request.form.get("text", "").strip()
+        if not original_text:
+            error_message = "Please enter some text to paraphrase."
+        else:
+            try:
+                paraphrased_texts = paraphrase_text(original_text)
+                if not paraphrased_texts:
+                    error_message = "Could not generate paraphrases. Please try different text."
+            except Exception as e:
+                error_message = f"An error occurred: {str(e)}"
+    return render_template(
+        "index.html",
+        paraphrased_texts=paraphrased_texts,
+        original_text=original_text,
+        error_message=error_message
+    )
+if __name__ == "__main__":
+    app.run(debug=True)

requirements.txt ADDED Viewed

	@@ -0,0 +1,5 @@

+fastapi
+uvicorn[standard]
+flask
+torch
+transformers

templates/index.html ADDED Viewed

	@@ -0,0 +1,89 @@

+<!-- templates/index.html -->
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI Paraphrasing Tool</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+            background-color: #f5f5f5;
+        }
+        .container {
+            background-color: white;
+            padding: 20px;
+            border-radius: 8px;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+        }
+        textarea {
+            width: 100%;
+            min-height: 100px;
+            margin: 10px 0;
+            padding: 10px;
+            border: 1px solid #ddd;
+            border-radius: 4px;
+            resize: vertical;
+        }
+        button {
+            background-color: #007bff;
+            color: white;
+            padding: 10px 20px;
+            border: none;
+            border-radius: 4px;
+            cursor: pointer;
+        }
+        button:hover {
+            background-color: #0056b3;
+        }
+        .error {
+            color: #dc3545;
+            margin: 10px 0;
+        }
+        .result {
+            margin-top: 20px;
+            padding: 15px;
+            background-color: #f8f9fa;
+            border-radius: 4px;
+        }
+        .variant {
+            margin: 10px 0;
+            padding: 10px;
+            background-color: #e9ecef;
+            border-radius: 4px;
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>AI Paraphrasing Tool</h1>
+        <form method="POST">
+            <div>
+                <label for="text">Enter text to paraphrase:</label>
+                <textarea name="text" id="text" required>{{ original_text }}</textarea>
+            </div>
+            <button type="submit">Generate Paraphrases</button>
+        </form>
+        {% if error_message %}
+        <div class="error">
+            {{ error_message }}
+        </div>
+        {% endif %}
+        {% if paraphrased_texts %}
+        <div class="result">
+            <h3>Paraphrased Versions:</h3>
+            {% for text in paraphrased_texts %}
+            <div class="variant">
+                {{ text }}
+            </div>
+            {% endfor %}
+        </div>
+        {% endif %}
+    </div>
+</body>
+</html>