Spaces:

0edon
/

KairosNews

Sleeping

Quintino Fernandes commited on May 16

Commit

e773b70

1 Parent(s): 94c2660

bug fix

Files changed (1) hide show

models/summarization.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 class SummarizationModel:
     def __init__(self):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.tokenizer = T5Tokenizer.from_pretrained('unicamp-dl/ptt5-base-portuguese-vocab').to(self.device)
         self.model = T5ForConditionalGeneration.from_pretrained('recogna-nlp/ptt5-base-summ').to(self.device)
     def summarize(self, text: str) -> str:

 class SummarizationModel:
     def __init__(self):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        self.tokenizer = T5Tokenizer.from_pretrained('unicamp-dl/ptt5-base-portuguese-vocab')
         self.model = T5ForConditionalGeneration.from_pretrained('recogna-nlp/ptt5-base-summ').to(self.device)
     def summarize(self, text: str) -> str: