AleRive commited on
Commit
690d3d8
·
verified ·
1 Parent(s): 1b5ee57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ os.environ["HF_HOME"] = "/tmp/huggingface"
12
  os.makedirs("/tmp/huggingface", exist_ok=True)
13
 
14
  # Carica il modello DialoGPT
15
- model_name = "microsoft/DialoGPT-small"
16
  tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir="/tmp/huggingface")
17
  model = AutoModelForCausalLM.from_pretrained(model_name, cache_dir="/tmp/huggingface")
18
 
 
12
  os.makedirs("/tmp/huggingface", exist_ok=True)
13
 
14
  # Carica il modello DialoGPT
15
+ model_name = "facebook/blenderbot-3B"
16
  tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir="/tmp/huggingface")
17
  model = AutoModelForCausalLM.from_pretrained(model_name, cache_dir="/tmp/huggingface")
18