diff --git a/recommendation_system.py b/recommendation_system.py index 3b90fff..20e730b 100644 --- a/recommendation_system.py +++ b/recommendation_system.py @@ -29,7 +29,7 @@ def recommendation(self, title=''): Returns: str: Dictionary with the movies recommended. """ - movie_index = self._df_movies[self._df_movies['title'].str.lower() == title].index.values + movie_index = self._df_movies[self._df_movies['title'].str.lower() == title.lower()].index.values if len(movie_index) > 0: similarity_scores = cosine_similarity(self._tfidf_matrix[movie_index], self._tfidf_matrix) diff --git a/requirements.txt b/requirements.txt index 96dda80..b6136da 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,7 +11,7 @@ python-dateutil pytz six sniffio -starlette +starlette==0.27.0 typing_extensions tzdata uvicorn