Hugging Face's logo

Garabatos
/
amigo

Model card Files
xet
 Community
Garabatos commited on
Commit
6ffa848
·
1 Parent(s): ac36ed4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
7
  import torch
8
 
9
  # ======== Cargar el modelo DialoGPT =========
10
- MODEL_NAME = "microsoft/DialoGPT-small"
11
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
12
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
13
 
 
7
  import torch
8
 
9
  # ======== Cargar el modelo DialoGPT =========
10
+ MODEL_NAME = "microsoft/DialoGPT-medium"
11
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
12
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
13