DavidHosp commited on
Commit
fc26408
1 Parent(s): 08816c7

update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -8
app.py CHANGED
@@ -26,14 +26,13 @@ def generar_recomendacion(svd_model, user_id, df, genres, top=5):
26
  recommended_titles = [id_to_title[movie_id] for movie_id, _ in recommended_movies[:top]]
27
 
28
  # Contar cuántas películas de cada género hay en las recomendaciones
29
- genre_counts = dfmerge[dfmerge['id'].isin([movie_id for movie_id, _ in recommended_movies])][genres].sum()
30
-
31
 
32
  # Limpiar la figura
33
  plt.clf()
34
 
35
  # Crear el gráfico de barras
36
- plt.bar(genres, genre_counts[genres])
37
  plt.xlabel('Género')
38
  plt.ylabel('Cantidad')
39
  plt.title('Cantidad de Películas por Género en las Recomendaciones')
@@ -51,9 +50,7 @@ def generar_recomendacion(svd_model, user_id, df, genres, top=5):
51
  # Devolver la lista de títulos y el gráfico como una imagen
52
  return ', '.join(recommended_titles), im
53
 
54
-
55
-
56
- # Leer los datos
57
  dfmerge = pd.read_csv('merged_data7.csv')
58
 
59
  # Cargar el modelo
@@ -94,5 +91,4 @@ Te mostraremos algunas películas que pueden gustarte.</b></p>
94
  )
95
 
96
  # Lanzar la interfaz
97
- demo.launch()
98
-
 
26
  recommended_titles = [id_to_title[movie_id] for movie_id, _ in recommended_movies[:top]]
27
 
28
  # Contar cuántas películas de cada género hay en las recomendaciones
29
+ genre_counts = df_filtered[df_filtered['id'].isin([movie_id for movie_id, _ in recommended_movies])][genres].sum()
 
30
 
31
  # Limpiar la figura
32
  plt.clf()
33
 
34
  # Crear el gráfico de barras
35
+ plt.bar(genres, genre_counts)
36
  plt.xlabel('Género')
37
  plt.ylabel('Cantidad')
38
  plt.title('Cantidad de Películas por Género en las Recomendaciones')
 
50
  # Devolver la lista de títulos y el gráfico como una imagen
51
  return ', '.join(recommended_titles), im
52
 
53
+ # Leer los datos
 
 
54
  dfmerge = pd.read_csv('merged_data7.csv')
55
 
56
  # Cargar el modelo
 
91
  )
92
 
93
  # Lanzar la interfaz
94
+ demo.launch()