dfedukov commited on
Commit
c7d91ee
·
verified ·
1 Parent(s): 769400a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -19,6 +19,9 @@ def load_model_and_tokenizer():
19
  return model, tokenizer
20
 
21
 
 
 
 
22
  def get_predict(title: str, abstract: str) -> (str, float, dict):
23
  tokenized_text = tokenizer(title + tokenizer.sep_token + abstact[:128], padding="max_length", truncation=True)
24
 
 
19
  return model, tokenizer
20
 
21
 
22
+ model, tokenizer = load_model_and_tokenizer()
23
+
24
+
25
  def get_predict(title: str, abstract: str) -> (str, float, dict):
26
  tokenized_text = tokenizer(title + tokenizer.sep_token + abstact[:128], padding="max_length", truncation=True)
27