Spaces:

unnat17
/

Text-Summarizer-Demo

Sleeping

unnat17 commited on about 1 month ago

Commit

e520358

verified ·

1 Parent(s): 119c97f

Upload app/app.py with huggingface_hub

Files changed (1) hide show

app/app.py CHANGED Viewed

@@ -27,7 +27,7 @@ app.mount(
 MODEL_ID = os.getenv("MODEL_ID", "unnat17/Text-Summarizer")
 model = T5ForConditionalGeneration.from_pretrained(MODEL_ID)
-tokenizer = T5Tokenizer.from_pretrained(MODEL_ID)
 if torch.cuda.is_available():
     device = torch.device("cuda")

 MODEL_ID = os.getenv("MODEL_ID", "unnat17/Text-Summarizer")
 model = T5ForConditionalGeneration.from_pretrained(MODEL_ID)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 if torch.cuda.is_available():
     device = torch.device("cuda")