Spaces:

govtech
/

lionguard-demo

Running

gabrielchua commited on Jun 23

Commit

6291db1

1 Parent(s): febc1d7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ model = LionGuard2()
 model.eval()
 # Load model weights
-model_path = os.path.join(os.path.dirname(__file__), '..', 'final_model', 'LionGuard2.safetensors')
 state_dict = load_file(model_path)
 model.load_state_dict(state_dict)

 model.eval()
 # Load model weights
+model_path = 'LionGuard2.safetensors'
 state_dict = load_file(model_path)
 model.load_state_dict(state_dict)