Update app.py

app.py

@@ -21,7 +21,7 @@ def run_test(model_type, repo_id, file_name, test_prompt, max_new_tokens, n_ctx,
     try:
         if model_type == "transformers":
             tokenizer = AutoTokenizer.from_pretrained(repo_id)
-            model = AutoModelForCausalLM.from_pretrained(repo_id)
+            model = AutoModelForCausalLM.from_pretrained(repo_id, trust_remote_code=True)
             inputs = tokenizer(test_prompt, return_tensors="pt")
             with torch.no_grad():
                 outputs = model.generate(