Edit model card

Model Card for Model ID

Baymax is a conversational AI model built specifically to help answer patient questions about health concerns. It’s based on Gemma-2B-IT but fine-tuned with real doctor-patient conversation data, so it’s trained to understand the context and sensitivities of medical discussions. The goal with Baymax is to create a model that’s not just informative, but also empathetic—providing answers that feel supportive and helpful, just like a real conversation with a healthcare provider.

Model Details

Model Description

This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.

  • Developed by: [Samarth Mishra]
  • Model type: [More Information Needed]
  • Language(s) (NLP): [More Information Needed]
  • License: [More Information Needed]
  • Finetuned from model [optional]: [google-gemma-2-2b-it]

Model Sources [optional]

  • Repository: [More Information Needed]
  • Paper [optional]: [More Information Needed]
  • Demo [optional]: [More Information Needed]

Uses

Recommendations

Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.

How to Get Started with the Model

Use the code below to get started with the model.

[More Information Needed]

Training Details

Training Data

[https://huggingface.co/datasets/lavita/ChatDoctor-HealthCareMagic-100k]

Downloads last month
12
Safetensors
Model size
2.61B params
Tensor type
FP16
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for samarth1029/Gemma-2-2b-baymax

Base model

google/gemma-2-2b
Finetuned
(116)
this model
Quantizations
2 models

Dataset used to train samarth1029/Gemma-2-2b-baymax