Mohammed-Altaf commited on
Commit
375fcf8
1 Parent(s): 9abc51e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -11,7 +11,7 @@ import torch
11
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
12
 
13
 
14
- path = "jianghc/medical_chatbot"
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
  tokenizer = GPT2Tokenizer.from_pretrained(path)
17
  model = GPT2LMHeadModel.from_pretrained(path).to(device)
 
11
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
12
 
13
 
14
+ path = "Mohammed-Altaf/Medical-ChatBot"
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
  tokenizer = GPT2Tokenizer.from_pretrained(path)
17
  model = GPT2LMHeadModel.from_pretrained(path).to(device)