mindwrapped commited on
Commit
a51b6ba
1 Parent(s): 53934f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ df["rating"] = df["rating"].values.astype(np.float32)
47
  # max_rating = max(df["rating"])
48
 
49
  # Load model
50
- model = from_pretrained_keras('mindwrapped/collaborative-filtering-movielens')
51
  movie_df = pd.read_csv(movielens_dir / "movies.csv")
52
 
53
 
 
47
  # max_rating = max(df["rating"])
48
 
49
  # Load model
50
+ model = from_pretrained_keras('keras-io/collaborative-filtering-movielens')
51
  movie_df = pd.read_csv(movielens_dir / "movies.csv")
52
 
53