sgonzalezsilot commited on
Commit
267ab6e
1 Parent(s): 3f3b7d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def bert_encode(tokenizer,data,maximum_length) :
41
 
42
  def get_news(input_text):
43
  sentence_length = 110
44
- train_input_ids,train_attention_masks = bert_encode(tokenizer,input_text,sentence_length)
45
 
46
  pred = m.predict([train_input_ids,train_attention_masks])
47
  pred = np.round(pred)
 
41
 
42
  def get_news(input_text):
43
  sentence_length = 110
44
+ train_input_ids,train_attention_masks = bert_encode(tokenizer,[input_text],sentence_length)
45
 
46
  pred = m.predict([train_input_ids,train_attention_masks])
47
  pred = np.round(pred)