Update app.py
Browse files
app.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
from huggingface_hub import InferenceClient
|
| 3 |
from transformers import pipeline
|
|
|
|
| 4 |
|
| 5 |
title = "Generador y clasificador de coloquialismos cubanos"
|
| 6 |
|
|
@@ -17,20 +18,22 @@ Se utiliz贸 el dataset somosnlp-hackathon-2026/patriae-cuba-literature-dataset p
|
|
| 17 |
* **Dionis Lopez (https://huggingface.co/inoid)**
|
| 18 |
"""
|
| 19 |
|
|
|
|
|
|
|
| 20 |
# Cargar el modelo para clasificaci贸n
|
| 21 |
classifier = pipeline(
|
| 22 |
"text-classification",
|
| 23 |
-
model='somosnlp-hackathon-2026/patriae-cuban-dialect-model'
|
|
|
|
| 24 |
)
|
| 25 |
|
| 26 |
def respond(message, history, system_message, max_tokens, temperature, top_p):
|
| 27 |
"""
|
| 28 |
Funci贸n para generar respuestas usando el modelo de chat
|
| 29 |
"""
|
| 30 |
-
# Usar el mismo modelo para generaci贸n (si es compatible)
|
| 31 |
-
|
| 32 |
client = InferenceClient(
|
| 33 |
-
model="somosnlp-hackathon-2026/patriae-cuban-dialect-model"
|
|
|
|
| 34 |
)
|
| 35 |
|
| 36 |
messages = [{"role": "system", "content": system_message}]
|
|
@@ -148,7 +151,7 @@ chat_interface = gr.ChatInterface(
|
|
| 148 |
label="Top-p (nucleus sampling)",
|
| 149 |
),
|
| 150 |
],
|
| 151 |
-
title="Chatbot con dialecto cubano
|
| 152 |
description="Genera texto con el dialecto cubano.",
|
| 153 |
examples=[
|
| 154 |
["隆Qu茅 bol谩, asere!"],
|
|
@@ -165,7 +168,7 @@ classifier_interface = gr.Interface(
|
|
| 165 |
lines=5
|
| 166 |
),
|
| 167 |
outputs=gr.Markdown(label="Resultado"),
|
| 168 |
-
title="
|
| 169 |
description="Detecta si un texto contiene expresiones coloquiales cubanas.",
|
| 170 |
examples=[
|
| 171 |
["隆Qu茅 bol谩, asere!"],
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
from huggingface_hub import InferenceClient
|
| 3 |
from transformers import pipeline
|
| 4 |
+
import os
|
| 5 |
|
| 6 |
title = "Generador y clasificador de coloquialismos cubanos"
|
| 7 |
|
|
|
|
| 18 |
* **Dionis Lopez (https://huggingface.co/inoid)**
|
| 19 |
"""
|
| 20 |
|
| 21 |
+
HF_TOKEN = os.environ.get("HF_TOKEN", None)
|
| 22 |
+
|
| 23 |
# Cargar el modelo para clasificaci贸n
|
| 24 |
classifier = pipeline(
|
| 25 |
"text-classification",
|
| 26 |
+
model='somosnlp-hackathon-2026/patriae-cuban-dialect-model',
|
| 27 |
+
token=HF_TOKEN
|
| 28 |
)
|
| 29 |
|
| 30 |
def respond(message, history, system_message, max_tokens, temperature, top_p):
|
| 31 |
"""
|
| 32 |
Funci贸n para generar respuestas usando el modelo de chat
|
| 33 |
"""
|
|
|
|
|
|
|
| 34 |
client = InferenceClient(
|
| 35 |
+
model="somosnlp-hackathon-2026/patriae-cuban-dialect-model",
|
| 36 |
+
token=HF_TOKEN
|
| 37 |
)
|
| 38 |
|
| 39 |
messages = [{"role": "system", "content": system_message}]
|
|
|
|
| 151 |
label="Top-p (nucleus sampling)",
|
| 152 |
),
|
| 153 |
],
|
| 154 |
+
title="Chatbot con dialecto cubano" ,
|
| 155 |
description="Genera texto con el dialecto cubano.",
|
| 156 |
examples=[
|
| 157 |
["隆Qu茅 bol谩, asere!"],
|
|
|
|
| 168 |
lines=5
|
| 169 |
),
|
| 170 |
outputs=gr.Markdown(label="Resultado"),
|
| 171 |
+
title="Clasificador de coloquialismos cubanos",
|
| 172 |
description="Detecta si un texto contiene expresiones coloquiales cubanas.",
|
| 173 |
examples=[
|
| 174 |
["隆Qu茅 bol谩, asere!"],
|