niknikita commited on
Commit
0b29129
1 Parent(s): 129f5a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ model = DistillBERTClass()
59
  LEARNING_RATE = 1e-05
60
 
61
  optimizer = torch.optim.Adam(params = model.parameters(), lr=LEARNING_RATE)
62
- model = torch.load("pytorch_distilbert_news.bin", map_location=torch.device('cpu'))
63
  # model.load_state_dict(checkpoint['model'])
64
  # optimizer.load_state_dict(checkpoint['opt'])
65
  # model.to("cpu")
 
59
  LEARNING_RATE = 1e-05
60
 
61
  optimizer = torch.optim.Adam(params = model.parameters(), lr=LEARNING_RATE)
62
+ model = torch.load("pytorch_distilbert_news (1).bin", map_location=torch.device('cpu'))
63
  # model.load_state_dict(checkpoint['model'])
64
  # optimizer.load_state_dict(checkpoint['opt'])
65
  # model.to("cpu")