Update app.py

app.py

@@ -17,8 +17,8 @@ system_message = "You are an AI assistant that specializes in analyzing and reas
 tabular_data = "col : day | kilometers row 1 : tuesday | 0 row 2 : wednesday | 0 row 3 : thursday | 4 row 4 : friday | 0 row 5 : saturday | 0"
 user_message = "Allie kept track of how many kilometers she walked during the past 5 days. What is the range of the numbers?"
 model_name = 'TIGER-Lab/StructLM-7B'
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto", trust_remote_code=True)
 # model.generation_config = GenerationConfig.from_pretrained(model_name)
 # model.generation_config.pad_token_id = model.generation_config.eos_token_id