Update app.py

app.py

@@ -21,6 +21,8 @@ def load_models():
   model_checkpoint = BertForTokenClassification.from_pretrained("dexay/Ner2HgF", )
    
   model_re = AutoModelForSequenceClassification.from_pretrained("dexay/reDs3others", )
+  
+  return tokenizer , model_checkpoint , model_re
 
 
 
@@ -36,9 +38,11 @@ if submit and len(x) != 0:
   #model.to("cpu")
   st.text("Execution is in progress ...")
   
-  load_models()
-  
+  tmr = load_models()
   
+  tokenizer = tmr[0]
+  model_checkpoint = tmr[1]
+  model_re = tmr[2]
   
   
   token_classifier = pipeline("token-classification", tokenizer = tokenizer,model=model_checkpoint,  )