jacksonstrut commited on
Commit
42db3a0
·
verified ·
1 Parent(s): 5ef32cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ if missing_vars:
30
  raise ValueError(f"Missing environment variables: {', '.join(missing_vars)}")
31
 
32
  # Initialize the Hugging Face tokenizer and model for DialoGPT
33
- model_name = "jacksonstrut/tinyllama-1.1B-chat"
34
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=HUGGINGFACE_API_TOKEN)
35
  model = AutoModelForCausalLM.from_pretrained(model_name, token=HUGGINGFACE_API_TOKEN)
36
  model.to('cpu')
 
30
  raise ValueError(f"Missing environment variables: {', '.join(missing_vars)}")
31
 
32
  # Initialize the Hugging Face tokenizer and model for DialoGPT
33
+ model_name = "Microsoft/DialoGPT-small"
34
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=HUGGINGFACE_API_TOKEN)
35
  model = AutoModelForCausalLM.from_pretrained(model_name, token=HUGGINGFACE_API_TOKEN)
36
  model.to('cpu')