Spaces:

Matthev00
/

Med_chat_bot

Sleeping

Matthev00 commited on Nov 19, 2023

Commit

5d39ec0

•

1 Parent(s): d6e5d75

working demo

Files changed (2) hide show

GPT_Model_med_bot.pth → Pretrained_GPT_med_bot.pth RENAMED Viewed

File without changes

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ def main():
     model = create_GPT_model(vocab_size=vocab_size, device=device)
     model.load_state_dict(torch.load(
-        f="Pretrianed_GPT_med_bot.pth",
         map_location=torch.device(device)))
     def predict(question: str):

     model = create_GPT_model(vocab_size=vocab_size, device=device)
     model.load_state_dict(torch.load(
+        f="Pretrained_GPT_med_bot.pth",
         map_location=torch.device(device)))
     def predict(question: str):