Update README.md

README.md

@@ -17,50 +17,40 @@ pipeline_tag: text-generation
 
 # Model Card for gpt-oss-20b-medical-qa
 
-This model is a fine-tuned version of [openai/gpt-oss-20b](https://huggingface.co/openai/gpt-oss-20b) on the [kingabzpro/gpt-oss-20b-medical-qa](https://huggingface.co/datasets/kingabzpro/gpt-oss-20b-medical-qa) dataset.
+This model is a fine-tuned version of [openai/gpt-oss-20b](https://huggingface.co/openai/gpt-oss-20b) on the [FreedomIntelligence/medical-o1-verifiable-problem](https://huggingface.co/datasets/FreedomIntelligence/medical-o1-verifiable-problem) dataset.
 It has been trained using [TRL](https://github.com/huggingface/trl).
 
 ## Quick start
 
 ```python
-from transformers import AutoModelForCausalLM, AutoTokenizer
-from peft import PeftModel
+from transformers import pipeline
 
-# Load the tokenizer
-tokenizer = AutoTokenizer.from_pretrained("openai/gpt-oss-20b")
+prompt = """<|start|>developer<|message|># Instructions
 
-# Load the original model first
-model_kwargs = dict(attn_implementation="eager", torch_dtype="auto", use_cache=True, device_map="auto")
-base_model = AutoModelForCausalLM.from_pretrained("openai/gpt-oss-20b", **model_kwargs).cuda()
+You are a medical expert with advanced knowledge in clinical reasoning and diagnostics. Respond with ONLY the final diagnosis/cause in ≤5 words.<|end|><|start|>user<|message|>An 88-year-old woman with osteoarthritis is experiencing mild epigastric discomfort and has vomited material resembling coffee grounds multiple times. Considering her use of naproxen, what is the most likely cause of her gastrointestinal blood loss?<|end|><|start|>assistant<|message|>"""
 
-# Merge fine-tuned weights with the base model
-peft_model_id = "kingabzpro/gpt-oss-20b-medical-qa"
-model = PeftModel.from_pretrained(base_model, peft_model_id)
-model = model.merge_and_unload()
-
-question = dataset[0]["Open-ended Verifiable Question"]
+# Load pipeline
+generator = pipeline(
+    "text-generation",
+    model="kingabzpro/gpt-oss-20b-medical-qa",
+    device="cuda"  # or device=0
+)
 
-text = render_infernce_harmony(question)
 
-inputs = tokenizer(
-    [text + tokenizer.eos_token], return_tensors="pt"
-).to("cuda")
-outputs = model.generate(
-    input_ids=inputs.input_ids,
-    attention_mask=inputs.attention_mask,
+# Run inference (passing in chat-style format)
+output = generator(
+    prompt,
     max_new_tokens=20,
-    eos_token_id=tokenizer.eos_token_id,
-    use_cache=True,
-)
-response = tokenizer.batch_decode(outputs)
-print(response[0])
+    return_full_text=False
+)[0]
+
+print(output["generated_text"])
+
 ```
 Output:
 
 ```bash
-<|start|>developer<|message|># Instructions
-
-You are a medical expert with advanced knowledge in clinical reasoning and diagnostics. Respond with ONLY the final diagnosis/cause in ≤5 words.<|end|><|start|>user<|message|>An 88-year-old woman with osteoarthritis is experiencing mild epigastric discomfort and has vomited material resembling coffee grounds multiple times. Considering her use of naproxen, what is the most likely cause of her gastrointestinal blood loss?<|end|><|start|>assistant<|return|><|message|>Stomach ulcer<|end|><|return|>
+NSAID use
 ```
 ## Training procedure
 This model was trained with SFT.