Blandskron commited on
Commit
d4f1b19
·
verified ·
1 Parent(s): f69941e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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 = [