update

app.py

@@ -6,8 +6,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model_tokenizer_path = "zehui127/Omni-DNA-DNA2Function"
 
 # Load tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained(model_tokenizer_path)
-model = AutoModelForCausalLM.from_pretrained(model_tokenizer_path).to('cuda')
+tokenizer = AutoTokenizer.from_pretrained(model_tokenizer_path,trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(model_tokenizer_path,trust_remote_code=True).to('cuda')
 
 def preprocess_response(response, mask_token="[MASK]"):
     """