Spaces:

tuongvxx1
/

medBot4

Running

tuongvxx1 commited on 3 days ago

Commit

618c89f

•

1 Parent(s): 760012a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ 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('medicalBot_ver4.pth', map_location=torch.device('cpu'))
 from collections import OrderedDict
 new_state_dict = OrderedDict()
 for k, v in state_dict.items():

 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'))
 from collections import OrderedDict
 new_state_dict = OrderedDict()
 for k, v in state_dict.items():