Downloads · 30 days
0
manishlang/gemma2-medical-llm
gemma2-medical-llm is a machine learning model from manishlang. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Gemma Medical Assistant is a fine-tuned LLM designed to provide health-related suggestions in both English and Hindi. It is based on Google Gemma-1.1-2B-IT and trained using QLoRA for lightweight, efficient medical gu…
Downloads · 30 days
0
Access
Public
Updated Nov 8, 2025
Repo size
41.1 MB
Likes
1
Public
Click a slice to open those files.
.json34.5 MB · 82%
From the Hugging Face model README
Gemma Medical Assistant is a fine-tuned LLM designed to provide health-related suggestions in both English and Hindi.
It is based on Google Gemma-1.1-2B-IT and trained using QLoRA for lightweight, efficient medical guidance responses.
✅ Bilingual: English + Hindi
✅ Medical symptom guidance
✅ Low-compute deployment
✅ Optimized for safety and simple explanations
| Property | Value |
|---|---|
| Base model | google/gemma-1.1-2b-it |
| Parameter efficient finetuning | QLoRA |
| LoRA Rank | 8 |
| Dataset size | 10,000 medical dialogue samples |
| Input format | Chat messages (messages: user + assistant) |
| Training hardware | Kaggle T4 GPU |
| Epochs | 2 |
| Max sequence length | 512 |
| Task | Causal Language Modeling / Chat |
| Objective | English–Hindi medical advice assistant |
This model is NOT a replacement for a doctor.
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch
model_id = "manishlang/gemma-medical"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
model_id,
torch_dtype=torch.float16,
device_map="auto"
)
prompt = "I have fever and cough. What should I do?"
inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
output = model.generate(**inputs, max_length=200)
print(tokenizer.decode(output[0], skip_special_tokens=True))