Update app.py
Browse files
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 = "
|
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 |
|