Downloads · 30 days
6
4% of all-time downloads
dkumarcvp/medical-llm-1gb
medical-llm-1gb is a text generation model from dkumarcvp. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
A medical-focused language model based on DistilGPT-2, optimized for healthcare applications.
Downloads · 30 days
6
4% of all-time downloads
All-time downloads
149
Public
Parameters
81.9M
328 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors328 MB · 99%
From the Hugging Face model README
A medical-focused language model based on DistilGPT-2, optimized for healthcare applications.
from transformers import AutoTokenizer, AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("dkumarcvp/medical-llm-1gb")
tokenizer = AutoTokenizer.from_pretrained("dkumarcvp/medical-llm-1gb")
# Generate medical text
prompt = "Question: What are the symptoms of diabetes?\nAnswer:"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_length=150, temperature=0.7)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(response)
⚠️ For educational and research purposes only. Not for medical diagnosis or treatment.