Update app.py

app.py

@@ -1,8 +1,7 @@
-import gradio as gr
-import os
 import warnings
 import random
 import torch
+
 from parrot import Parrot
 
 # Suppress warnings
@@ -10,16 +9,13 @@ warnings.filterwarnings("ignore")
 os.environ['TRANSFORMERS_NO_ADVISORY_WARNINGS'] = '1'
 
 # Set random state for reproducibility
-torch.manual_seed(1234)
-if torch.cuda.is_available():
+def random_state(seed):
+    torch.manual_seed(seed)
+    if torch.cuda.is_available():
     torch.cuda.manual_seed_all(1234)
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-print(f"Using device: {device}")
-
 
 # Initialize Parrot model
-parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")
-parrot.model.to(device)   
+parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")     
 # Function to paraphrase a single sentence
 def paraphrase_sentence(sentence):
     paraphrases = parrot.augment(input_phrase=sentence, max_return_phrases=10, max_length=100, adequacy_threshold=0.75, fluency_threshold=0.75)