clavel commited on
Commit
cf422eb
verified
1 Parent(s): f55fff3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -6
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="馃攳 Clasificador de coloquialismos cubanos",
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!"],