Spaces:

niknikita
/

transformerYSDA

Runtime error

niknikita commited on Apr 16, 2023

Commit

5ba71e5

•

1 Parent(s): 3be3c4a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,7 +63,7 @@ optimizer = torch.optim.Adam(params =  model.parameters(), lr=LEARNING_RATE)
-model = torch.load("pytorch_distilbert_news (4).bin", map_location=torch.device('cpu'))
 def get_predict(title, abstract):


63
64
65
66	+ model = torch.load("bert_distilbert.bin", map_location=torch.device('cpu'))
67
68
69	def get_predict(title, abstract):