Update app.py

app.py

@@ -14,7 +14,7 @@ app = FastAPI()
 # Check for GPU and load model accordingly
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_name = os.getenv("MODEL_NAME")
-model = SentenceTransformer(model_name, device=device)
+model = SentenceTransformer(model_name, device=device, trust_remote_code=True)
 
 # Define the embedding endpoint
 @app.post("/embed")