Blandskron commited on
Commit
e19d06a
·
verified ·
1 Parent(s): 9c2b25a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,12 +2,12 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
2
  import gradio as gr
3
 
4
  # Cargar modelo y tokenizador desde Hugging Face Hub
5
- repo_id = "Blandskron/ModelChatBot" # Nombre del repositorio en Hugging Face Hub
6
 
7
  try:
8
  model = AutoModelForCausalLM.from_pretrained(
9
  repo_id,
10
- ignore_mismatched_sizes=True # Ignorar desajuste de tamaños
11
  )
12
  tokenizer = AutoTokenizer.from_pretrained(repo_id)
13
  except Exception as e:
 
2
  import gradio as gr
3
 
4
  # Cargar modelo y tokenizador desde Hugging Face Hub
5
+ repo_id = "Blandskron/ModelChatBot"
6
 
7
  try:
8
  model = AutoModelForCausalLM.from_pretrained(
9
  repo_id,
10
+ ignore_mismatched_sizes=True
11
  )
12
  tokenizer = AutoTokenizer.from_pretrained(repo_id)
13
  except Exception as e: