Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 import difflib
 
 # Create a Parrot instance
-parrot = Parrot(model_tag="turing", use_gpu=False)
+parrot = Parrot(model_tag="t5-small", use_gpu=False)
 
 # Supported languages for paraphrasing
 languages = {
@@ -18,8 +18,9 @@ languages = {
 
 # Supported models for paraphrasing
 models = {
-    "Turing": "turing",
-    "Pegasus": "pegasus"
+    "T5 Small": "t5-small",
+    "T5 Base": "t5-base",
+    "Pegasus": "google/pegasus-multi_news"
 }
 
 def paraphrase_text(text, language, model, level):
@@ -52,21 +53,21 @@ interface = gr.Interface(
         gr.inputs.Slider(minimum=1, maximum=10, default=5, label="Level of paraphrasing:"),
         gr.inputs.Checkbox(label="Generate audio output?"),
         gr.inputs.File(label="Upload file containing text to paraphrase (Optional):")
-],
-outputs=[
-gr.outputs.Textbox(label="Paraphrased text:"),
-gr.outputs.Download(label="Download paraphrased text as text file:"),
-gr.outputs.HTML(label="Highlighted differences:")
-],
-title="Parrot Paraphrasing Tool",
-description="Enter text to paraphrase, select language and model, and choose level of paraphrasing."
+    ],
+    outputs=[
+        gr.outputs.Textbox(label="Paraphrased text:"),
+        gr.outputs.Download(label="Download paraphrased text as text file:"),
+        gr.outputs.HTML(label="Highlighted differences:")
+    ],
+    title="Parrot Paraphrasing Tool",
+    description="Enter text to paraphrase, select language and model, and choose level of paraphrasing."
 )
 
 def app(text, language, model, level, audio_output, file_upload):
     if file_upload:
         with open(file_upload.name, "r", encoding="utf-8") as f:
             text = f.read()
-            paraphrased_text = paraphrase_text(text, languages[language], models[model], level)
+    paraphrased_text = paraphrase_text(text, languages[language], models[model], level)
     if audio_output:
         speak_paraphrased_text(paraphrased_text, languages[language])
         download_output = gr.outputs.Download(paraphrased_text, label="Download paraphrased text as text file:", mime_type="text/plain")