clovershield_ml_api / seed_model.py
GitHub Action
Sync from GitHub: 06d29d1048a570c62cc184ddba97f3e1bd3640a2
01242e9
Raw
History Blame Contribute Delete
1.02 kB
import os
from supabase import create_client
from dotenv import load_dotenv
load_dotenv()
# Initialize Supabase
url = os.environ.get("SUPABASE_URL")
# using Service Role Key to bypass RLS if needed, or simple ANON key if policy allows
key = os.environ.get("SUPABASE_SERVICE_ROLE_KEY") or os.environ.get("SUPABASE_KEY")
if not url or not key:
print("Error: Supabase credentials not found in environment.")
exit(1)
supabase = create_client(url, key)
model_data = {
"name": "Production Model (Initial)",
"version": "v1.0.0",
"status": "ready",
"metrics": {
"accuracy": 0.998,
"precision": 0.908,
"recall": 1.00,
"f1": 0.952,
"specificity": 0.998
},
"file_path": "Models/fraud_pipeline_final.pkl",
"is_active": True
}
try:
data = supabase.table("model_registry").insert(model_data).execute()
print("Successfully registered existing model:")
print(data.data)
except Exception as e:
print(f"Failed to register model: {e}")