Spaces:

tuongvxx1
/

medBot4_2

Running

tuongvxx1 commited on 3 days ago

Commit

21776ed

•

1 Parent(s): 8865786

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-base")
 model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-base")
-model_file = hf_hub_download(repo_id="tuongvxx1/medBot", filename="medicalBot_ver4.pth")
 state_dict = torch.load(model_file, map_location=torch.device('cpu'))
 model.load_state_dict(new_state_dict)
 model.to(device)

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-base")
 model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-base")
+model_file = hf_hub_download(repo_id="tuongvxx1/medBot", filename="medicalBot_ver4_2.pth")
 state_dict = torch.load(model_file, map_location=torch.device('cpu'))
 model.load_state_dict(new_state_dict)
 model.to(device)