Spaces:

VietCat
/

ViT5BaseNode

Sleeping

VietCat commited on Jun 11

Commit

b9e0b8c

1 Parent(s): 4159c4a

switch model to VietAI/vit5-base-vietnews-summarization

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 app = FastAPI()
 # Load model và tokenizer
-model_name = "VietAI/vit5-base"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 app = FastAPI()
 # Load model và tokenizer
+model_name = "VietAI/vit5-base-vietnews-summarization"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")