Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"
|
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:
|
|
|
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:
|