README.md

metadata

title: FunctionGemma Tuning Lab
short_description: How to fine-tune FunctionGemma for tool calling
emoji: 📊
colorFrom: gray
colorTo: indigo
sdk: gradio
sdk_version: 6.3.0
app_file: app.py
pinned: false
hf_oauth: true
hf_oauth_scopes:
  - manage-repos
license: apache-2.0

FunctionGemma Tuning Lab

FunctionGemma Tuning Lab is a user-friendly, Gradio-based interface demo designed to help you fine-tune FunctionGemma model to understand and utilize your specific custom tools and functions.

Whether you are building an agent to query internal knowledge bases, control smart home devices, or interact with proprietary APIs, this lab streamlines the process of teaching the model your specific function schemas.

Features

• Interactive Tool Definition: Define your function schemas (JSON) directly in the UI.
• Custom Dataset Import: Upload your own training data via CSV.
• One-Click Fine-Tuning: Configure hyperparameters (Epochs, Learning Rate) and start training with a single click.
• Real-Time Monitoring: Watch training logs and loss curves update in real-time.
• Automatic Evaluation: The system automatically evaluates the model's performance before and after training to show improvement.
• Export Artifacts: Download your fine-tuned model weights ready for deployment.

Installation

1. Clone the repository:

   hf download google/functiongemma-tuning-lab --repo-type=space --local-dir=functiongemma-tuning-lab
   cd functiongemma-tuning-lab
   

2. Install dependencies: It is recommended to use a virtual environment.

   pip install -r requirements.txt
   

Usage

1. Set up Environment Variables (Optional): If you need to access gated models on Hugging Face, set your token:

   export HF_TOKEN=your_huggingface_token
   

2. Run the Application:

   python app.py
   

3. Access the UI: Open your browser and navigate to the local URL provided in the terminal (usually http://127.0.0.1:7860).

Data Format

To train on your own data, upload a CSV file with the following columns (no header required, or header ignored if present):

1. User Prompt: The natural language query from the user.
2. Tool Name: The name of the function that should be called.
3. Tool Arguments: A JSON string representing the arguments for the function.

Example CSV Row:

"What is the weather in London?", "get_weather", "{""location"": ""London, UK""}"

Configuration

Core settings can be modified in config.py:

• MODEL_NAME: Base model to fine-tune (default: google/functiongemma-270m-it).
• DEFAULT_DATASET: Hugging Face dataset to use if no custom CSV is uploaded.
• ARTIFACTS_DIR: Directory where training outputs are saved.