Update app.py

app.py

@@ -6,9 +6,15 @@ import gradio as gr
 from src.app.response import describe_video
 
 
-# Video and text inputs for the interface
+# Video, text query, and input parameters
 video = gr.Video(label="Video")
 query = gr.Textbox(label="Question", placeholder="Enter your question here")
+temperature = gr.Slider(
+    minimum=0.01, maximum=1.99, step=0.01, value=0.7, label="Temperature"
+)
+top_p = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.8, label="Top P")
+top_k = gr.Slider(minimum=0, maximum=1000, step=1, value=100, label="Top K")
+max_new_tokens = gr.Slider(minimum=1, maximum=4096, step=1, value=512, label="Max Tokens")
 
 # Output for the interface
 response = gr.Textbox(label="Predicted answer", show_label=True, show_copy_button=True)
@@ -18,12 +24,27 @@ examples = [
     [
         "./videos/sample_video_1.mp4",
         "Here are some frames of a video. Describe this video in detail.",
+        0.7,
+        0.8,
+        100,
+        512,
     ],
     [
         "./videos/sample_video_2.mp4",
         "¿Cuál es el animal de este vídeo? ¿Cuantos animales hay?",
+        0.7,
+        0.8,
+        100,
+        512,
+    ],
+    [
+        "./videos/sample_video_3.mp4",
+        "Que se passe-t-il dans cette vidéo ?",
+        0.7,
+        0.8,
+        100,
+        512,
     ],
-    ["./videos/sample_video_3.mp4", "Que se passe-t-il dans cette vidéo ?"],
 ]
 
 # Title, description, and article for the interface
@@ -35,7 +56,7 @@ article = "<p style='text-align: center'><a href='https://github.com/OpenBMB/Min
 # Launch the interface
 interface = gr.Interface(
     fn=describe_video,
-    inputs=[video, query],
+    inputs=[video, query, temperature, top_p, top_k, max_new_tokens],
     outputs=response,
     examples=examples,
     cache_examples=True,