Views
No views yet
1from mlx_lm import load, generate
2
3model, tokenizer = load("Cognitapp/Cognitapp-Med-Nano-v1")
4prompt = "<system>You are the Cognitapp Global ICD-10 Assistant. Extract the primary ICD-10 code.</system> <user>Patient has 103F fever, body aches, and positive NS1 for Dengue.</user> <assistant>"
5response = generate(model, tokenizer, prompt=prompt, max_tokens=10)
6print(response)