gpu update

app.py

@@ -10,16 +10,16 @@ warnings.filterwarnings("ignore")
 os.environ['TRANSFORMERS_NO_ADVISORY_WARNINGS'] = '1'
 
 # Set random state for reproducibility
-def random_state(seed):
-    torch.manual_seed(seed)
-    if torch.cuda.is_available():
-        torch.cuda.manual_seed_all(seed)
+torch.manual_seed(1234)
+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}")
 
-random_state(1234)
 
 # Initialize Parrot model
 parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")
-
+parrot.model.to(device)   
 # 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)