Upload app.py with huggingface_hub

app.py

@@ -1,5 +1,5 @@
 """
-Axiom-Ref — HuggingFace Space / Gradio App
+Axiom — HuggingFace Space / Gradio App
 
 Two modes:
   1. GENERATE — produce governed output with proof
@@ -38,9 +38,9 @@ from pipeline.stages.s4_validate import validate_and_score, TigStatus
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 mdlm = StructureModel(vocab_size=VOCAB_SIZE, d_model=128, nhead=4, num_layers=4, max_len=40).to(device)
-mdlm.load_state_dict(torch.load("models/axiom-ref/mdlm_best.pt", weights_only=True, map_location=device))
+mdlm.load_state_dict(torch.load("models/axiom/mdlm_best.pt", weights_only=True, map_location=device))
 
-tokenizer = Tokenizer.from_file("models/axiom-ref/bpe_tokenizer.json")
+tokenizer = Tokenizer.from_file("models/axiom/bpe_tokenizer.json")
 bpe_vocab = tokenizer.get_vocab_size()
 BPE_BOS = tokenizer.token_to_id("<bos>")
 BPE_EOS = tokenizer.token_to_id("<eos>")
@@ -49,7 +49,7 @@ decoder = ConstrainedDecoder(
     gov_vocab=VOCAB_SIZE, prose_vocab=bpe_vocab, d_model=256, nhead=8,
     num_encoder_layers=3, num_decoder_layers=6, max_struct_len=40, max_prose_len=128,
 ).to(device)
-_ds = torch.load("models/axiom-ref/decoder_best.pt", weights_only=True, map_location=device)
+_ds = torch.load("models/axiom/decoder_best.pt", weights_only=True, map_location=device)
 _ds = {k.replace("triad_embedding", "struct_embedding").replace("triad_pos", "struct_pos"): v for k, v in _ds.items()}
 decoder.load_state_dict(_ds)
 decoder.eval()
@@ -246,12 +246,12 @@ def verify_governance(output_text, trace_json_str):
 # ═══════════════════════════════════════════════════════════════════════════════
 
 with gr.Blocks(
-    title="Axiom-Ref: Governed Language Model",
+    title="Axiom: Governed Language Model",
     theme=gr.themes.Base(primary_hue="green", neutral_hue="slate"),
 ) as app:
 
     gr.Markdown("""
-    # Axiom-Ref
+    # Axiom
     **Every output ships its own proof of governance.**
     """)
 
@@ -305,7 +305,7 @@ with gr.Blocks(
 
     gr.Markdown("""
     ---
-    *[MetaCortex Dynamics DAO](https://github.com/MetaCortex-Dynamics) · [Source](https://github.com/MetaCortex-Dynamics/Axiom-Ref) · MIT License*
+    *[MetaCortex Dynamics DAO](https://github.com/MetaCortex-Dynamics) · [Source](https://github.com/MetaCortex-Dynamics/Axiom) · MIT License*
     """)
 
 if __name__ == "__main__":