PunchNFIT commited on
Commit
b69d24b
·
1 Parent(s): 8d22a8e

Fix requirements and add trust_remote_code

Browse files
Files changed (1) hide show
  1. api_inference.py +1 -1
api_inference.py CHANGED
@@ -14,7 +14,7 @@ print(f"🔍 Loading model from {MODEL_DIR} ...")
14
  try:
15
  config = AutoConfig.from_pretrained(MODEL_DIR)
16
  tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
17
- model = AutoModel.from_pretrained(MODEL_DIR, config=config)
18
  model.eval()
19
  print("✅ SNP Universal Embedding model loaded successfully.")
20
  except Exception as e:
 
14
  try:
15
  config = AutoConfig.from_pretrained(MODEL_DIR)
16
  tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
17
+ model = AutoModel.from_pretrained(MODEL_DIR, config=config, trust_remote_code=True)
18
  model.eval()
19
  print("✅ SNP Universal Embedding model loaded successfully.")
20
  except Exception as e: