Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -191,7 +191,7 @@ def get_user_emb(embs, ys):
|
|
191 |
print('ys are longer than embs; popping latest rating')
|
192 |
ys.pop(-1)
|
193 |
|
194 |
-
feature_embs = np.array(torch.stack([embs[i].squeeze().to('cpu') for i in indices]
|
195 |
#scaler = preprocessing.StandardScaler().fit(feature_embs)
|
196 |
#feature_embs = scaler.transform(feature_embs)
|
197 |
chosen_y = np.array([ys[i] for i in indices])
|
|
|
191 |
print('ys are longer than embs; popping latest rating')
|
192 |
ys.pop(-1)
|
193 |
|
194 |
+
feature_embs = np.array(torch.stack([embs[i].squeeze().to('cpu') for i in indices]).to('cpu'))
|
195 |
#scaler = preprocessing.StandardScaler().fit(feature_embs)
|
196 |
#feature_embs = scaler.transform(feature_embs)
|
197 |
chosen_y = np.array([ys[i] for i in indices])
|