Spaces:
Running
Running
File size: 1,148 Bytes
7609276 c17dbac 146289a 9265924 146289a a233401 b8c9b58 a233401 d4399fb b8c9b58 7609276 c17dbac d4399fb b8c9b58 c17dbac b8c9b58 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
from flask import Flask, request, render_template
import transformers
from parrot import Parrot
import torch
parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")
#from flask import Flask, request, render_template
#import parrot
#import transformers
import time
app = Flask(__name__)
@app.route("/")
def index():
return render_template("index.html")
@app.route("/paraphrase", methods=["POST"])
def paraphrase():
text = request.form["text"]
model = request.form["model"]
if model == "parrot":
start_time = time.time()
paraphrased_text = parrot.augment(input_phrase=text, use_gpu=False)
processing_time = time.time() - start_time
elif model == "transformers":
start_time = time.time()
paraphrased_text = transformers.paraphrase(text)
processing_time = time.time() - start_time
processing_percentage = 100 * processing_time / 30
return render_template("paraphrase.html", original_text=text, paraphrased_text=paraphrased_text, processing_percentage=processing_percentage)
if __name__ == "__main__":
app.run(host="0.0.0.0",port=7860) |