update

app.py

@@ -8,7 +8,10 @@ import spaces
 import cv2
 import numpy as np
 from PIL import Image
-from models import get_model_list, get_model_info, DEFAULT_GENERATION_PARAMS
+from models import (
+    get_model_list, get_model_info, DEFAULT_GENERATION_PARAMS,
+    get_preset_list, get_preset_params, get_preset_description
+)
 
 def progress_bar_html(label: str) -> str:
     """
@@ -228,8 +231,26 @@ examples = [
 ]
 
 def create_interface():
-    # Get the list of available models
+    # Get the list of available models and presets
     model_options = get_model_list()
+    preset_options = get_preset_list()
+    
+    def apply_preset(preset_name):
+        """Helper function to apply parameter presets"""
+        params = get_preset_params(preset_name)
+        return [
+            params["temperature"],
+            params["top_p"],
+            params["top_k"],
+            params["max_new_tokens"],
+            params["do_sample"],
+            params["num_beams"],
+            params["early_stopping"],
+            params["length_penalty"],
+            params["no_repeat_ngram_size"],
+            params["repetition_penalty"],
+            get_preset_description(preset_name)
+        ]
     
     with gr.Blocks() as demo:
         gr.Markdown("# **Qwen2.5 Series (add `@video-infer` for video understanding)**")
@@ -242,6 +263,21 @@ def create_interface():
                     label="Select Model"
                 )
             
+            with gr.Row():
+                preset_dropdown = gr.Dropdown(
+                    choices=preset_options,
+                    value="Default",
+                    label="Parameter Preset"
+                )
+                preset_description = gr.Textbox(
+                    value=get_preset_description("Default"),
+                    label="Preset Description",
+                    interactive=False
+                )
+            
+            # Button to apply the selected preset
+            preset_button = gr.Button("Apply Preset")
+            
             with gr.Row():
                 temperature = gr.Slider(
                     minimum=0.0,

models.py

@@ -105,3 +105,42 @@ def get_model_list():
         list: List of model names
     """
     return list(RECOMMENDED_MODELS.keys())
+
+def get_preset_list():
+    """
+    Returns a list of available parameter presets.
+    
+    Returns:
+        list: List of preset names
+    """
+    return list(GENERATION_PRESETS.keys())
+
+def get_preset_params(preset_name):
+    """
+    Returns the generation parameters for a given preset.
+    
+    Args:
+        preset_name (str): Name of the preset
+        
+    Returns:
+        dict: Generation parameters
+    """
+    return GENERATION_PRESETS.get(preset_name, DEFAULT_GENERATION_PARAMS)
+
+def get_preset_description(preset_name):
+    """
+    Returns a description for the given preset.
+    
+    Args:
+        preset_name (str): Name of the preset
+        
+    Returns:
+        str: Description of the preset
+    """
+    descriptions = {
+        "Default": "Balanced parameters suitable for most use cases",
+        "Creative": "Higher temperature and diversity for more creative outputs",
+        "Precise": "Lower temperature with beam search for more precise, focused responses",
+        "Deterministic": "Greedy decoding with beam search for deterministic, consistent outputs"
+    }
+    return descriptions.get(preset_name, "Custom parameters")