Removed tensor.

app.py

@@ -7,22 +7,17 @@ import spaces
 from sumy.nlp.tokenizers import Tokenizer
 from sumy.parsers.plaintext import PlaintextParser
 from sumy.summarizers.text_rank import TextRankSummarizer
-import torch
 from transformers import PegasusForConditionalGeneration, PegasusTokenizer
 
 nltk.download('punkt_tab')
 
 
-zero = torch.Tensor([0]).cuda()
-print(zero.device)
-
 class ModelLoader:
     def __init__(self):
         self.model_checkpoint = 'AryehRotberg/Pegasus-Large-Privacy-Policy-Summarization-V2'
         self.model = PegasusForConditionalGeneration.from_pretrained(self.model_checkpoint).to('cuda')
         self.tokenizer = PegasusTokenizer.from_pretrained(self.model_checkpoint)
     
-    @spaces.GPU
     def predict(self, text):
         inputs = self.tokenizer(
             f'Summarize the following document: {text}\nSummary: ',