DavidHosp commited on
Commit
f722506
·
1 Parent(s): a2578a5

update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -47,6 +47,13 @@ def generar_recomendacion(svd_model, user_id, df, genres, top=5):
47
  # Devolver la lista de títulos y el gráfico como una imagen
48
  return ', '.join(recommended_titles), im
49
 
 
 
 
 
 
 
 
50
  # Modificar la función wrap_generar_recomendacion para devolver una imagen también
51
  def wrap_generar_recomendacion(user_id, drama, comedy, horror, romance, top=5):
52
  # Crear la lista de géneros de interés a partir de las casillas de verificación
 
47
  # Devolver la lista de títulos y el gráfico como una imagen
48
  return ', '.join(recommended_titles), im
49
 
50
+ # Leer los datos
51
+ dfmerge = pd.read_csv('merged_data6.csv')
52
+
53
+ # Cargar el modelo
54
+ with open('fc_model_svd_v1.pkl', 'rb') as file:
55
+ svd_model = pickle.load(file)
56
+
57
  # Modificar la función wrap_generar_recomendacion para devolver una imagen también
58
  def wrap_generar_recomendacion(user_id, drama, comedy, horror, romance, top=5):
59
  # Crear la lista de géneros de interés a partir de las casillas de verificación