Memory release

app.py

@@ -15,8 +15,9 @@ MAX_LEN = 256
 MODEL = BertAD()
 vec = MODEL.state_dict()['bert.embeddings.position_ids']
 chkp = torch.load(os.path.join('model', 'model_0.bin'), map_location='cpu')
-chkp['bert.embeddings.position_ids'] =vec
+chkp['bert.embeddings.position_ids'] = vec
 MODEL.load_state_dict(chkp)
+del chkp, vec
 
 
 def sample_text(text, acronym, max_len):

model.py

@@ -6,7 +6,7 @@ class BertAD(nn.Module):
   def __init__(self):
     super(BertAD, self).__init__()
 
-    model_config = transformers.AutoConfig.from_pretrained('model')
+    model_config = transformers.AutoConfig.from_pretrained('./model')
     model_config.update({"output_hidden_states":True})
 
     self.bert = transformers.BertModel(model_config)