Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -70,8 +70,9 @@ def obtener_predicciones(texto, nombre_modelo, top_k=10):
|
|
70 |
return top_k_tokens, top_k_prob.cpu().tolist()
|
71 |
|
72 |
def generar_barplot(tokens, probabilidades):
|
73 |
-
"""Genera una
|
74 |
-
datos = {"Palabra":
|
|
|
75 |
return datos
|
76 |
|
77 |
def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
|
@@ -134,10 +135,10 @@ with gr.Blocks() as demo:
|
|
134 |
|
135 |
with gr.Row():
|
136 |
barplot_resultados = gr.BarPlot(
|
137 |
-
|
138 |
x="Palabra",
|
139 |
y="Probabilidad",
|
140 |
-
title="Predicciones del modelo"
|
141 |
)
|
142 |
|
143 |
# Acciones de botones
|
|
|
70 |
return top_k_tokens, top_k_prob.cpu().tolist()
|
71 |
|
72 |
def generar_barplot(tokens, probabilidades):
|
73 |
+
"""Genera una lista de diccionarios para Gradio BarPlot."""
|
74 |
+
datos = [{"Palabra": t, "Probabilidad": p} for t, p in zip(tokens, probabilidades)]
|
75 |
+
print(datos)
|
76 |
return datos
|
77 |
|
78 |
def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
|
|
|
135 |
|
136 |
with gr.Row():
|
137 |
barplot_resultados = gr.BarPlot(
|
138 |
+
value=[], # Se inicializa vacío
|
139 |
x="Palabra",
|
140 |
y="Probabilidad",
|
141 |
+
title="📊 Predicciones del modelo"
|
142 |
)
|
143 |
|
144 |
# Acciones de botones
|