alcaxel commited on
Commit
1809da3
·
verified ·
1 Parent(s): ab629b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py CHANGED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ # Cargar el modelo tsundere de Hugging Face
5
+ natsuki = pipeline("text-generation", model="ericzhou/tsundere_v1")
6
+
7
+ # Función para generar respuestas de Natsuki
8
+ def chat_fn(message):
9
+ if not message.strip(): # Si no se escribió nada o es solo espacio
10
+ return "¿No vas a decir nada? ¡Habla, tonto!"
11
+ # Generar respuesta usando el modelo en español
12
+ res = natsuki(f"Tú: {message}\nNatsuki:", max_new_tokens=100, do_sample=True, temperature=0.7)
13
+ reply = res[0]["generated_text"].split("Natsuki:")[-1].strip() # Limpiar el texto
14
+ # Traducir la respuesta a español si es necesario
15
+ if reply:
16
+ return reply
17
+ else:
18
+ return "No sé qué decir, ¡pero no te hagas el tonto!"
19
+
20
+ # Crear la interfaz gráfica usando Gradio
21
+ iface = gr.Interface(fn=chat_fn,
22
+ inputs=gr.Textbox(lines=2, placeholder="Escribe algo..."),
23
+ outputs="text",
24
+ title="Natsuki-Bot",
25
+ description="Tu waifu tsundere, ¡que hable y sea spicy en español!",
26
+ allow_flagging="never")
27
+
28
+ iface.launch()