Spaces:

ANWAR101
/

Recommendation

Sleeping

Anwar11234 commited on Jul 3

Commit

382ebba

•

1 Parent(s): d823dd2

fixed error in recommender.py

Files changed (1) hide show

recommender.py CHANGED Viewed

@@ -26,7 +26,7 @@ def recommend(course_data: dict = Body(...)):
   # Get the course indices
   course_indices = [i[0] for i in sim_scores]
-  recommendations = df.iloc[course_indices][['CourseID', 'Title']].to_dict(orient='records')
   return recommendations

   # Get the course indices
   course_indices = [i[0] for i in sim_scores]
+  recommendations = df.iloc[course_indices][['Id', 'Title']].to_dict(orient='records')
   return recommendations