Spaces:

sussahoo
/

item_reco_lstm

Build error

sussahoo commited on Dec 14, 2022

Commit

ffe82b4

•

1 Parent(s): 4a892b0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -119,7 +119,7 @@ norm_type = int(setup_config["norm_type"])
 max_norm = int(setup_config["max_norm"])
 model = Net( vocab_size, embedding_dim, norm_type, max_norm, window_size)
-model.load_state_dict(torch.load("/content/sample_data/model_lstm.pth"))
 model.eval()

 max_norm = int(setup_config["max_norm"])
 model = Net( vocab_size, embedding_dim, norm_type, max_norm, window_size)
+model.load_state_dict(torch.load("model_lstm.pth"))
 model.eval()