Update app.py

app.py

@@ -13,13 +13,13 @@ os.environ['TRANSFORMERS_NO_ADVISORY_WARNINGS'] = '1'
 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}")
+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.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)