Update README.md

README.md

@@ -13,10 +13,14 @@ tags:
 - medgemma
 pipeline_tag: text-generation
 widget:
-- text: "USER: What are the common treatments for type 2 diabetes?\nASSISTANT:"
+- text: |-
+    USER: What are the common treatments for type 2 diabetes?
+    ASSISTANT:
+base_model:
+- google/medgemma-4b-it
 ---
 
-# Model Card: MedGemma-4B-Abliterated-Finetuned-[YourVersionName]
+# Model Card: MedGemma-4B-Abliterated
 
 ## Model Description
 
@@ -81,4 +85,4 @@ full_prompt = prompt_template.format(instruction=instruction)
 
 inputs = tokenizer(full_prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(**inputs, max_new_tokens=200, do_sample=True, temperature=0.7, top_p=0.9)
-print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))