Update app.py

app.py

@@ -325,23 +325,27 @@ import gradio as gr
 import pickle
 import pandas as pd
 import numpy as np
+import os
+
+# Define base directory
+BASE_DIR = 'deployment_files' if os.path.exists('deployment_files') else '.'
 
-# Load all models
-with open('user_cf_model.pkl', 'rb') as f:
+# Load all models with correct paths
+with open(f'{BASE_DIR}/user_cf_model.pkl', 'rb') as f:
     user_cf = pickle.load(f)
 
-with open('item_cf_model.pkl', 'rb') as f:
+with open(f'{BASE_DIR}/item_cf_model.pkl', 'rb') as f:
     item_cf = pickle.load(f)
 
-with open('svd_model.pkl', 'rb') as f:
+with open(f'{BASE_DIR}/svd_model.pkl', 'rb') as f:
     svd = pickle.load(f)
 
-with open('user_item_matrix.pkl', 'rb') as f:
+with open(f'{BASE_DIR}/user_item_matrix.pkl', 'rb') as f:
     user_item_matrix = pickle.load(f)
 
-movies = pd.read_csv('movies.csv')
+movies = pd.read_csv(f'{BASE_DIR}/movies.csv')
 
-with open('metrics.pkl', 'rb') as f:
+with open(f'{BASE_DIR}/metrics.pkl', 'rb') as f:
     metrics = pickle.load(f)
 
 MODELS = {