Kiran2004 commited on
Commit
8750abf
·
verified ·
1 Parent(s): 5fb0d8f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -41,7 +41,7 @@ The model's performance can be evaluated using various metrics such as F1 score,
41
  from transformers import pipeline, GPT2Tokenizer, GPT2LMHeadModel
42
 
43
  # Load tokenizer and model
44
- model_name = "Kiran2004/GPT2_Conversational_Bot"
45
  tokenizer = GPT2Tokenizer.from_pretrained(model_name)
46
  model = GPT2LMHeadModel.from_pretrained(model_name)
47
 
 
41
  from transformers import pipeline, GPT2Tokenizer, GPT2LMHeadModel
42
 
43
  # Load tokenizer and model
44
+ model_name = "Kiran2004/GPT2_MentalHealth_ChatBot"
45
  tokenizer = GPT2Tokenizer.from_pretrained(model_name)
46
  model = GPT2LMHeadModel.from_pretrained(model_name)
47