Spaces:

niknikita
/

transformerYSDA

Runtime error

niknikita commited on Apr 16, 2023

Commit

1da12a6

•

1 Parent(s): 238ca26

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ model = torch.load("pytorch_distilbert_news (4).bin", map_location=torch.device(
 def get_predict(title, abstract):
     tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-cased')
-    inputs = tokenizer(title, abstract,  return_tensors="pt")
     outputs = model(
         input_ids=inputs['input_ids'],
         attention_mask=inputs['attention_mask'],

 def get_predict(title, abstract):
     tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-cased')
+    inputs = tokenizer(abstract,  title, return_tensors="pt")
     outputs = model(
         input_ids=inputs['input_ids'],
         attention_mask=inputs['attention_mask'],