Update README.md

README.md

@@ -59,12 +59,27 @@ pip install torch transformers soundfile torchaudio
 ### Inference
 
 ```python
-from transformers import AutoModel
-
-model   = AutoModel.from_pretrained("amaai-lab/apex", trust_remote_code=True)
-results = model.predict("my_song.mp3", save_json="results.json")
-
-print(results["score_streams"])  # popularity score 0-100
-print(results["score_likes"])    # popularity score 0-100
-print(results["coherence"])      # aesthetic score 1-5
+formers import AutoModel
+import torch
+
+model = AutoModel.from_pretrained(
+    "amaai-lab/apex",
+    trust_remote_code       = True,
+    device_map              = None,
+    low_cpu_mem_usage       = False,
+    ignore_mismatched_sizes = True
+)
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model  = model.to(device)
+
+results = model.predict("/path/to/your/mp3/file", save_json="results.json")
+
+print(f"Streams Score : {results['score_streams']:.2f}")
+print(f"Likes Score   : {results['score_likes']:.2f}")
+print(f"Coherence     : {results['coherence']:.2f}")
+print(f"Musicality    : {results['musicality']:.2f}")
+print(f"Memorability  : {results['memorability']:.2f}")
+print(f"Clarity       : {results['clarity']:.2f}")
+print(f"Naturalness   : {results['naturalness']:.2f}")
 ```