joao-victor-campos NicholasBaraldi commited on
Commit
eb2d08f
1 Parent(s): c6e75fa

Co-authored-by: NicholasBaraldi <nicholas.baraldi@gmail.com>

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from recommendation_app.core.data_handler.data_handler import DataHandler
4
  from recommendation_app.core.model import Model
5
 
6
  PATH = "data/output/df_titles.csv"
7
- df2 = pd.read_csv(PATH)
8
- movie_names = df2["title"].tolist()
9
 
10
 
11
  def gradio(movie_name: str, n_rec: int) -> pd.DataFrame:
 
4
  from recommendation_app.core.model import Model
5
 
6
  PATH = "data/output/df_titles.csv"
7
+ df_recommendation = pd.read_csv(PATH)
8
+ movie_names = df_recommendation["title"].tolist()
9
 
10
 
11
  def gradio(movie_name: str, n_rec: int) -> pd.DataFrame: