igenesissbot / app.py
cesarniantic's picture
Update app.py
f78f52f verified
raw
history blame contribute delete
633 Bytes
import gradio as gr
from transformers import pipeline
# Cargar Mistral-7B (versión ligera)
generator = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta")
def responder(mensaje):
prompt = f"Sos un argentino porteño, usá 'vos', 'che', 'boludo' y jerga rioplatense. Respondé a: '{mensaje}'"
respuesta = generator(prompt, max_length=100, num_return_sequences=1)[0]["generated_text"]
return respuesta.split("Respondé a:")[1].strip() if "Respondé a:" in respuesta else respuesta
# Interfaz simple
interfaz = gr.Interface(fn=responder, inputs="text", outputs="text", title="igenesiss")
interfaz.launch()