Spaces:

tuongvxx1
/

medBot4_2

Running

tuongvxx1 commited on 3 days ago

Commit

3d48c86

•

1 Parent(s): 21776ed

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_2.pth")
 state_dict = torch.load(model_file, map_location=torch.device('cpu'))
-model.load_state_dict(new_state_dict)
 model.to(device)
 def generate_answer(question, model, tokenizer, device):

 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(state_dict)
 model.to(device)
 def generate_answer(question, model, tokenizer, device):