Spaces:
Running
Running
File size: 924 Bytes
7609276 c17dbac fd93e78 9265924 fd93e78 a233401 d4399fb c17dbac 7609276 c17dbac 9265924 c17dbac 7609276 c17dbac d4399fb c17dbac |
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 |
from flask import Flask, request, render_template
import transformers
import parrot
import torch
#parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")
app = Flask(__name__)
@app.route("/", methods=["GET", "POST"])
def index():
if request.method == "POST":
text = request.form["text"]
library = request.form["library"]
if library == "transformers":
model = transformers.AutoModelWithLMHead.from_pretrained("gpt2")
paraphrased_text = transformers.generate(model, text)
elif library == "parrot":
paraphrased_text = parrot.augment(input_phrase=text, use_gpu=False)
else:
return "Error: Invalid library selected"
return render_template("index.html", original_text=text, paraphrased_text=paraphrased_text)
return render_template("index.html")
if __name__ == "__main__":
app.run(host="0.0.0.0",port=7860)
|