Emuixom commited on
Commit
3b85876
verified
1 Parent(s): 4de610e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,12 +1,14 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- # Cargar el modelo de DeepSeek AI para el chatbot (esto depende de c贸mo se cargue tu modelo)
5
  @st.cache_resource
6
  def load_model():
7
  # Aseg煤rate de que el nombre del modelo es correcto o ajusta seg煤n el que uses
8
  model_name = "deepseek-ai/DeepSeek-R1" # Cambia esto si el modelo es diferente
9
- chatbot = pipeline("conversational", model=model_name)
 
 
10
  return chatbot
11
 
12
  # Inicializa el modelo
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ # Cargar el modelo de DeepSeek AI para el chatbot
5
  @st.cache_resource
6
  def load_model():
7
  # Aseg煤rate de que el nombre del modelo es correcto o ajusta seg煤n el que uses
8
  model_name = "deepseek-ai/DeepSeek-R1" # Cambia esto si el modelo es diferente
9
+
10
+ # Agregar trust_remote_code=True para permitir la ejecuci贸n del c贸digo remoto
11
+ chatbot = pipeline("conversational", model=model_name, trust_remote_code=True)
12
  return chatbot
13
 
14
  # Inicializa el modelo