Update app.py

app.py

@@ -6,6 +6,9 @@ from utils import *
 HengamTransW = hf_hub_download(repo_id="kargaranamir/Hengam", filename="HengamTransW.pth")
 HengamTransA = hf_hub_download(repo_id="kargaranamir/Hengam", filename="HengamTransA.pth")
 
+ner_W = NER(model_path=HengamTransW, tags=['B-TIM', 'I-TIM', 'B-DAT', 'I-DAT', 'O'])
+ner_A = NER(model_path=HengamTransA, tags=['B-TIM', 'I-TIM', 'B-DAT', 'I-DAT', 'O'])
+
 
 # APP
 st.set_page_config(
@@ -57,10 +60,9 @@ with st.form(key="my_form"):
         )
 
         if ModelType == 'HengamTransW.pth':
-            ModelPath = HengamTransW
+            ner = ner_W
         else:
-            ModelPath = HengamTransA
-        ner = NER(model_path=ModelPath, tags=['B-TIM', 'I-TIM', 'B-DAT', 'I-DAT', 'O'])
+            ner = ner_A
 
     with c2:
         doc = st.text_area(