Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,10 +1,38 @@
|
|
| 1 |
-
import
|
|
|
|
| 2 |
|
| 3 |
-
#
|
| 4 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
# Si tienes un modelo en HuggingFace, se puede cargar como este ejemplo
|
| 6 |
+
@st.cache_resource
|
| 7 |
+
def load_model():
|
| 8 |
+
# Aseg煤rate de que el nombre del modelo es correcto o ajusta seg煤n el que uses
|
| 9 |
+
model_name = "deepseek-ai/DeepSeek-R1" # Cambia esto si el modelo es diferente
|
| 10 |
+
chatbot = pipeline("conversational", model=model_name)
|
| 11 |
+
return chatbot
|
| 12 |
|
| 13 |
+
# Inicializa el modelo
|
| 14 |
+
chatbot = load_model()
|
| 15 |
+
|
| 16 |
+
def main():
|
| 17 |
+
# T铆tulo de la aplicaci贸n
|
| 18 |
+
st.title("ChatBot - DeepSeek AI")
|
| 19 |
+
|
| 20 |
+
# Introducci贸n
|
| 21 |
+
st.write("隆Bienvenido al ChatBot! Puedes hacerle preguntas y recibir respuestas.")
|
| 22 |
+
|
| 23 |
+
# Campo de entrada de texto
|
| 24 |
+
user_input = st.text_input("Escribe tu pregunta:", "")
|
| 25 |
+
|
| 26 |
+
if user_input:
|
| 27 |
+
# Generar una respuesta del chatbot usando el modelo
|
| 28 |
+
response = chatbot(user_input)
|
| 29 |
+
|
| 30 |
+
# Mostrar la respuesta generada
|
| 31 |
+
st.write(f"**Respuesta del chatbot:** {response[0]['generated_text']}")
|
| 32 |
+
|
| 33 |
+
# Bot贸n para reiniciar el chat (opcional)
|
| 34 |
+
if st.button("Reiniciar chat"):
|
| 35 |
+
st.write("Chat reiniciado. Puedes comenzar de nuevo.")
|
| 36 |
+
|
| 37 |
+
if __name__ == "__main__":
|
| 38 |
+
main()
|