Update app.py

app.py

@@ -1,6 +1,8 @@
 from flask import Flask, request, render_template
 import transformers
-import parrot
+from parrot import Parrot
+import torch
+parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")
 
 app = Flask(__name__)
 
@@ -14,7 +16,7 @@ def index():
             model = transformers.AutoModelWithLMHead.from_pretrained("gpt2")
             paraphrased_text = transformers.generate(model, text)
         elif library == "parrot":
-            paraphrased_text = parrot.paraphrase(text)
+            paraphrased_text = parrot.augment(input_phrase=text, use_gpu=False)
         else:
             return "Error: Invalid library selected"