14B

app.py

@@ -145,8 +145,10 @@ def process_video(
     global DKT_PIPELINE
 
     if model_size == "14B":
+        logger.info(f'14B model is chosen')
         pipeline = DKT_PIPELINE_14B
     elif model_size == "1.3B":
+        logger.info(f'1.3B model is chosen')
         pipeline = DKT_PIPELINE
     else:
         raise ValueError(f"Invalid model size: {model_size}")
@@ -448,7 +450,7 @@ with gr.Blocks(css=css, title="DKT", head=head_html) as demo:
         
     def on_example_submit(video_file):
         """Wrapper function for examples with default parameters"""
-        return on_submit(video_file, "1.3B", 5, 3)
+        return on_submit(video_file, "14B", 5, 3)
 
     examples = gr.Examples(
         examples=example_inputs,