Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,11 +2,10 @@ import json
|
|
2 |
import gradio as gr
|
3 |
from transformers import pipeline
|
4 |
|
5 |
-
# Paso 1: Usar Hugging Face Pipeline para el modelo
|
6 |
MODEL_NAME = "deepseek-ai/DeepSeek-R1"
|
7 |
|
8 |
# Usar el pipeline de Hugging Face para modelos de lenguaje causal
|
9 |
-
model = pipeline("text-generation", model=MODEL_NAME, tokenizer=MODEL_NAME)
|
10 |
|
11 |
# Paso 2: Definir la estructura JSON de prueba
|
12 |
data = [
|
|
|
2 |
import gradio as gr
|
3 |
from transformers import pipeline
|
4 |
|
|
|
5 |
MODEL_NAME = "deepseek-ai/DeepSeek-R1"
|
6 |
|
7 |
# Usar el pipeline de Hugging Face para modelos de lenguaje causal
|
8 |
+
model = pipeline("text-generation", model=MODEL_NAME, tokenizer=MODEL_NAME, trust_remote_code=True)
|
9 |
|
10 |
# Paso 2: Definir la estructura JSON de prueba
|
11 |
data = [
|