sussahoo commited on
Commit
498eb73
1 Parent(s): ab4510e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -97,13 +97,13 @@ class Net(nn.Module):
97
 
98
  return logits, h, c
99
 
100
- with open("/content/sample_data/setup_config.json") as f:
101
  setup_config = json.load(f)
102
 
103
- with open("/content/sample_data/ch2ix.json") as f:
104
  ch2ix = json.load(f)
105
 
106
- with open("/content/sample_data/vocab.json") as f:
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("/content/sample_data/model_lstm.pth"))
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