Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|