teaevo commited on
Commit
3380313
·
1 Parent(s): f24bed6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
20
 
21
  def chatbot_response(user_message):
22
  # Load the pre-trained model and tokenizer
23
- model_name = "your_pretrained_model_name" # Replace with the name of the pre-trained model you want to use
24
  tokenizer = AutoTokenizer.from_pretrained(model_name)
25
  model = AutoModelForCausalLM.from_pretrained(model_name)
26
 
 
20
 
21
  def chatbot_response(user_message):
22
  # Load the pre-trained model and tokenizer
23
+ model_name = "gpt2" # Replace with the name of the pre-trained model you want to use
24
  tokenizer = AutoTokenizer.from_pretrained(model_name)
25
  model = AutoModelForCausalLM.from_pretrained(model_name)
26