Spaces:

niknikita
/

transformerYSDA

Runtime error

niknikita commited on Apr 16, 2023

Commit

d02d7c5

1 Parent(s): 72228c4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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


77
78
79
80	+ model = torch.load("pytorch_distilbert_news (4).bin", map_location=torch.device('cpu'))
81
82
83	def get_predict(title, abstract):