stal76 commited on
Commit
2dc13a4
·
1 Parent(s): aa213c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ class Net(nn.Module):
20
  return self.layer(x)
21
 
22
  model = Net()
23
- model.load_state_dict(torch.load('model.dat'), map_location=torch.device('cpu'))
24
  tokenizer = AutoTokenizer.from_pretrained("Callidior/bert2bert-base-arxiv-titlegen")
25
  model_emb = AutoModelForSeq2SeqLM.from_pretrained("Callidior/bert2bert-base-arxiv-titlegen")
26
 
 
20
  return self.layer(x)
21
 
22
  model = Net()
23
+ model.load_state_dict(torch.load('model.dat', map_location=torch.device('cpu')))
24
  tokenizer = AutoTokenizer.from_pretrained("Callidior/bert2bert-base-arxiv-titlegen")
25
  model_emb = AutoModelForSeq2SeqLM.from_pretrained("Callidior/bert2bert-base-arxiv-titlegen")
26