update app.py
Browse files
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
|