CoolFace
Apppublic

satyanayak/GRPO-Training

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes
App README

Gemma-3 Model Comparison Demo

This is a Hugging Face Spaces demo application that showcases and compares the base Gemma-3 model with its GRPO-finetuned version. The application provides an interactive interface to generate and compare text from both models side by side.

Features

  • Side-by-side comparison of base and finetuned models
  • Interactive text generation for both models
  • Example prompts to get started
  • Clean and intuitive Gradio interface
  • Real-time comparison of model outputs

Usage

  1. 1.Enter your prompt in either text box (or use the same prompt for both)
  2. 2.Click the respective "Generate" button for each model
  3. 3.Compare the outputs from both models
  4. 4.Try the example prompts to see different types of responses

Model Information

The demo uses two models:

Technical Details

  • Built with Gradio
  • Uses PyTorch for model inference
  • Optimized with float16 precision for better performance
  • Parallel processing of both models
  • Automatic device mapping for GPU/CPU

Example Prompts

The demo includes example prompts that showcase the differences between the base and finetuned models:

  • Mathematical questions
  • Counting tasks
  • Word problems

These examples help demonstrate the improvements in the GRPO-finetuned model's responses.