Update app.py

app.py

@@ -155,7 +155,7 @@ def tweets_localidad(buscar_localidad):
             muestra = st.table(df.reset_index(drop=True).head(10).style.applymap(color_survived, subset=['Prediccion']))
             tabla.append(muestra)
             #resultado=df.groupby('Prediccion')['Probabilidad'].sum()
-            resultado = df['Prediccion'].value_counts()
+            resultado = df['Prediccion'].head(10).value_counts()
             colores=["#EE3555","#aae977"]
             fig, ax = plt.subplots()
             fig.set_size_inches(2, 2)
@@ -168,15 +168,15 @@ def tweets_localidad(buscar_localidad):
             st.set_option('deprecation.showPyplotGlobalUse', False)
             st.pyplot()
             
-            #plt.bar(resultado.index, resultado, color=colores)
-            #ax.set_title("Porcentajes por Categorias", fontsize=5, fontweight="bold")
-            #plt.rcParams.update({'font.size':4, 'font.weight':'bold'})
-            #ax.set_xlabel("Categoría")
-            #ax.set_ylabel("Probabilidad")
+            plt.bar(resultado.index, resultado, color=colores)
+            ax.set_title("Porcentajes por Categorias", fontsize=5, fontweight="bold")
+            plt.rcParams.update({'font.size':4, 'font.weight':'bold'})
+            ax.set_xlabel("Categoría")
+            ax.set_ylabel("Probabilidad")
             # Muestra el gráfico
-            #plt.show()
-            #st.set_option('deprecation.showPyplotGlobalUse', False)
-            #st.pyplot()
+            plt.show()
+            st.set_option('deprecation.showPyplotGlobalUse', False)
+            st.pyplot()
         
     except AttributeError as e:
         muestra=st.text("No existe ninguna localidad con ese nombre")