Spaces:

Curative
/

t5-small_cnn

Sleeping

Curative commited on Apr 27

Commit

f95bb71

verified ·

1 Parent(s): 94695da

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 # Load your fine-tuned model
 model_path = "./t5-summarizer"  # Path inside Docker container
 model = T5ForConditionalGeneration.from_pretrained(model_path)
-tokenizer = T5Tokenizer.from_pretrained(model_path)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)

 # Load your fine-tuned model
 model_path = "./t5-summarizer"  # Path inside Docker container
 model = T5ForConditionalGeneration.from_pretrained(model_path)
+tokenizer = T5Tokenizer.from_pretrained(model_path, legacy=False)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)