Model Card for Gemma Pharmacy💊
Model Details
Model Description
Gemma Pharmacy helps users find the best OTC medication for common symptoms such as headaches, indigestion, and more. The model was fine-tuned on a dataset that includes detailed medication information, making it an effective tool for anyone looking to manage their health by selecting the right medication. Users benefit from quick access to critical medication details, including side effects and dosage guidelines. 🤗
- Developed by: Sooyeon Jang, Noa Jeong
- Shared by: Sooyeon Jang, Noa Jeong
- Model type: Transformer-based model for symptom-to-medication recommendation
- Language(s) (NLP): Korean
- Finetuned from model: Hugging Face Gemma model
Model Sources [optional]
- Repository: [More Information Needed]
- Paper [optional]: [More Information Needed]
- Demo [optional]: [More Information Needed]
Uses
Direct Use
This model can be used directly by users looking to identify OTC medications based on their symptoms.
Downstream Use
Fine-tuning on specific symptoms, expanding to more languages, or integrating into health applications.
Out-of-Scope Use
Not intended for use in critical medical diagnoses or treatment suggestions where professional healthcare advice is necessary.
Bias, Risks, and Limitations
This model was fine-tuned on a Korean medication dataset and may not generalize well to non-Korean OTC medications or medical contexts. Additionally, the model’s recommendations should not be considered as professional medical advice, and users should always consult a healthcare provider.
Recommendations
Users should cross-check the recommendations provided by the model with certified medical professionals, especially in cases of serious symptoms or chronic conditions.
How to Get Started with the Model
Use the code below to get started with the model.
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
model_name = "gemmagemma/gemma-pharmacy"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
Training Details
Training Data
The 'e약은요' dataset contains information on OTC medications, including their efficacy, usage methods, and side effects. This dataset was used to fine-tune the Gemma Pharmacy model.
Training Procedure
Training Hyperparameters
- Training regime: fp16 mixed precision
Evaluation
Results
The model achieves strong performance in predicting correct medications based on symptoms, particularly for common conditions such as headaches and indigestion.
Environmental Impact
Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).
- Hardware Type: Nvidia GPU
Technical Specifications
Model Architecture and Objective
The model is based on a transformer architecture with the objective of symptom-to-medication recommendation.
- Downloads last month
- 22