Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -97,13 +97,13 @@ class Net(nn.Module):
|
|
97 |
|
98 |
return logits, h, c
|
99 |
|
100 |
-
with open("
|
101 |
setup_config = json.load(f)
|
102 |
|
103 |
-
with open("
|
104 |
ch2ix = json.load(f)
|
105 |
|
106 |
-
with open("
|
107 |
vocab = json.load(f)
|
108 |
|
109 |
window_size = int(setup_config["window_size"])
|
@@ -113,7 +113,7 @@ norm_type = int(setup_config["norm_type"])
|
|
113 |
max_norm = int(setup_config["max_norm"])
|
114 |
|
115 |
model = Net( vocab_size, embedding_dim, norm_type, max_norm, window_size)
|
116 |
-
model.load_state_dict(torch.load("
|
117 |
model.eval()
|
118 |
|
119 |
|
|
|
97 |
|
98 |
return logits, h, c
|
99 |
|
100 |
+
with open("setup_config.json") as f:
|
101 |
setup_config = json.load(f)
|
102 |
|
103 |
+
with open("ch2ix.json") as f:
|
104 |
ch2ix = json.load(f)
|
105 |
|
106 |
+
with open("vocab.json") as f:
|
107 |
vocab = json.load(f)
|
108 |
|
109 |
window_size = int(setup_config["window_size"])
|
|
|
113 |
max_norm = int(setup_config["max_norm"])
|
114 |
|
115 |
model = Net( vocab_size, embedding_dim, norm_type, max_norm, window_size)
|
116 |
+
model.load_state_dict(torch.load("model_lstm.pth"))
|
117 |
model.eval()
|
118 |
|
119 |
|