Upload app.py

app.py

@@ -16,13 +16,13 @@ try:
 except:
     print("english model load error")
 
+'''
 try:
-    tokenizer_multilingual = AutoTokenizer.from_pretrained("amir22010/PyABSA_Hospital_Multilingual_allenai_tk-instruct-base-def-pos_FinedTuned_Model")
+    tokenizer_multilingual = AutoTokenizer.from_pretrained("amir22010/amir22010/PyABSA_Hospital_Multilingual_allenai_tk-instruct-base-def-pos_FinedTuned_Model")
     double_multilingual_generator = AutoModelForSeq2SeqLM.from_pretrained("amir22010/PyABSA_Hospital_Multilingual_allenai_tk-instruct-base-def-pos_FinedTuned_Model")
 except:
     print("multilingual model load error")
 
-'''
 try:
     tokenizer_keybert = AutoTokenizer.from_pretrained("amir22010/KeyBert_ABSA_Hospital_Multilingual_allenai_tk-instruct-base-def-pos_FinedTuned_Model")
     double_keybert_generator = AutoModelForSeq2SeqLM.from_pretrained("amir22010/KeyBert_ABSA_Hospital_Multilingual_allenai_tk-instruct-base-def-pos_FinedTuned_Model")
@@ -35,9 +35,9 @@ def perform_asde_inference(text, dataset, model_id):
     if not text:
         if model_id == "PyABSA_Hospital_English_allenai/tk-instruct-base-def-pos_FinedTuned_Model":
             df = pd.read_csv('pyabsa_english.csv')#validation dataset
+        '''
         elif model_id == "PyABSA_Hospital_Multilingual_allenai/tk-instruct-base-def-pos_FinedTuned_Model":
             df = pd.read_csv('pyabsa_multilingual.csv')#validation dataset
-        '''
         elif model_id == "KeyBert_ABSA_Hospital_allenai/tk-instruct-base-def-pos_FinedTuned_Model":
             df = pd.read_csv('keybert_valid.csv')#validation dataset
         '''
@@ -67,11 +67,12 @@ def perform_asde_inference(text, dataset, model_id):
         output = double_english_generator.generate(tokenized_text.input_ids,max_length=512)
         model_generated = tokenizer_english.decode(output[0], skip_special_tokens=True)
 
+    '''
     elif model_id == "PyABSA_Hospital_Multilingual_allenai/tk-instruct-base-def-pos_FinedTuned_Model":
         tokenized_text = tokenizer_multilingual(bos_instruction + text + delim_instruct + eos_instruct, return_tensors="pt")
         output = double_multilingual_generator.generate(tokenized_text.input_ids,max_length=512)
         model_generated = tokenizer_multilingual.decode(output[0], skip_special_tokens=True)
-    '''
+    
     elif model_id == "KeyBert_ABSA_Hospital_allenai/tk-instruct-base-def-pos_FinedTuned_Model":
         tokenized_text = tokenizer_keybert(bos_instruction + text + delim_instruct + eos_instruct, return_tensors="pt")
         output = double_keybert_generator.generate(tokenized_text.input_ids,max_length=512)
@@ -123,7 +124,7 @@ if __name__ == "__main__":
                         asde_model_ids = gr.Radio(
                             choices=[
                                 "PyABSA_Hospital_English_allenai/tk-instruct-base-def-pos_FinedTuned_Model", 
-                                "PyABSA_Hospital_Multilingual_allenai/tk-instruct-base-def-pos_FinedTuned_Model", 
+                                #"PyABSA_Hospital_Multilingual_allenai/tk-instruct-base-def-pos_FinedTuned_Model", 
                                 #"KeyBert_ABSA_Hospital_allenai/tk-instruct-base-def-pos_FinedTuned_Model"
                             ],
                             value="PyABSA_Hospital_English_allenai/tk-instruct-base-def-pos_FinedTuned_Model",