Mohammed-Altaf
/

Medical-ChatBot

Text Generation

text-generation-inference

Model card Files Files and versions

Mohammed-Altaf commited on Nov 20, 2023

Commit

375fcf8

·

1 Parent(s): 9abc51e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ import torch
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
-path = "jianghc/medical_chatbot"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 tokenizer = GPT2Tokenizer.from_pretrained(path)
 model = GPT2LMHeadModel.from_pretrained(path).to(device)

 from transformers import GPT2LMHeadModel, GPT2Tokenizer
+path = "Mohammed-Altaf/Medical-ChatBot"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 tokenizer = GPT2Tokenizer.from_pretrained(path)
 model = GPT2LMHeadModel.from_pretrained(path).to(device)