Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import PegasusForConditionalGeneration, PegasusTokenizer
 model_name = 'tuner007/pegasus_paraphrase'
 # torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
 tokenizer = PegasusTokenizer.from_pretrained(model_name)
-model = PegasusForConditionalGeneration.from_pretrained(model_name).to(torch_device)
+model = PegasusForConditionalGeneration.from_pretrained(model_name)
 
 def paraphrase(text):
     from sentence_splitter import SentenceSplitter, split_text_into_sentences