FunctionGemma-Director-V1

🎬 FunctionGemma-Director-V1

FunctionGemma-Director-V1 is a specialized lightweight AI model (270M parameters) designed to automate the production of viral short-form gaming videos (TikTok/Shorts/Reels).

It acts as a "Creative Director", converting a simple video title into a structured JSON editing plan, executing a "Trojan Horse" monetization strategy by seamlessly integrating CPA offers into content.

🚀 Key Features

• Size: ~540MB (Runs smoothly on free Colab/CPU).
• Strategy: Automatically places "High Retention Hooks" and injects "CPA Offers" at the most effective timestamps.
• Output: Strict JSON format compatible with Python video automation engines (MoviePy).

💻 How to Use

import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
import json

# 1. Load the Model
model_id = "Saad4web/FunctionGemma-Director-V1"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id, 
    device_map="auto", 
    torch_dtype=torch.float16 # Optimized for low memory
)

# 2. Define the Tools (The Model's Vocabulary)
tools_schema = [
    {"name": "add_video_clip", "parameters": {"file_path": "string", "duration": "number"}},
    {"name": "add_text_overlay", "parameters": {"text": "string", "color": "string"}},
]

# 3. Create the Prompt
video_title = "TOP 3|SCARIEST HORROR GAMES|*DONT WATCH ALONE*"
system_msg = f"You are a specialized video editor AI. Available tools: {json.dumps(tools_schema)}"
messages = [{"role": "user", "content": system_msg + f"\n\nCreate a viral video plan for: {video_title}"}]

# 4. Generate the Plan
input_ids = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)

outputs = model.generate(
    input_ids, 
    max_new_tokens=512, 
    do_sample=True, 
    temperature=0.1
)

# 5. Get the JSON
plan = tokenizer.decode(outputs[0][len(input_ids[0]):], skip_special_tokens=True)
print(plan)

#🛠️ Training Details
    Architecture: Fine-tuned google/functiongemma-270m-it.
    Dataset: Synthetic dataset generated via Knowledge Distillation (Teacher: GPT-4o/Gemini 2.0).
    Method: Full Fine-Tuning using LLaMA Factory.
    Created by [Saad4web]