Update app.py

app.py

@@ -6,11 +6,9 @@ from transformers import BartTokenizer, BartForConditionalGeneration, AutoModel,
 bart_tokenizer = BartTokenizer.from_pretrained('facebook/bart-large')
 bart_model = BartForConditionalGeneration.from_pretrained('webshop/il_search_bart')
 
-if False:  # TODO: make sure it could be uploaded from hub
-    bert_tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased', truncation_side='left')
-    bert_tokenizer.add_tokens(['[button]', '[button_]', '[clicked button]', '[clicked button_]'], special_tokens=True)
-    bert_model = AutoModel.from_pretrained('webshop/il_choice_bert')
-
+bert_tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased', truncation_side='left')
+bert_tokenizer.add_tokens(['[button]', '[button_]', '[clicked button]', '[clicked button_]'], special_tokens=True)
+bert_model = AutoModel.from_pretrained('webshop/il-choice-bert-image_0', trust_remote_code=True)
 
 def process_str(s):
     s = s.lower().replace('"', '').replace("'", "").strip()