Update app.py

app.py

@@ -28,10 +28,13 @@ footer {
 
 # Functions
 @spaces.GPU(duration=60)
-def generate(image, instruction=DEFAULT_INPUT, sampling=False, temperature=0.7, top_p=0.8, top_k=100, repetition_penalty=1.05, max_tokens=512):
+def generate(image, video, instruction=DEFAULT_INPUT, sampling=False, temperature=0.7, top_p=0.8, top_k=100, repetition_penalty=1.05, max_tokens=512):
     repo.to(DEVICE)
 
     print(image)
+    print(video)
+    print(not video)
+    
     image_data = Image.fromarray(image.astype('uint8'), 'RGB')
     print(image_data, instruction)
 
@@ -60,6 +63,7 @@ with gr.Blocks(css=css) as main:
         
     with gr.Column():
         input = gr.Image(label="Image")
+        input_2 = gr.Image(label="Video")
         instruction = gr.Textbox(lines=1, value=DEFAULT_INPUT, label="Instruction")
         sampling = gr.Checkbox(value=False, label="Sampling")
         temperature = gr.Slider(minimum=0.01, maximum=1.99, step=0.01, value=0.7, label="Temperature")
@@ -73,7 +77,7 @@ with gr.Blocks(css=css) as main:
     with gr.Column():
         output = gr.Textbox(lines=1, value="", label="Output")
 
-    submit.click(fn=generate, inputs=[input, instruction, sampling, temperature, top_p, top_k, repetition_penalty, max_tokens], outputs=[output], queue=False)
+    submit.click(fn=generate, inputs=[input, input_2, instruction, sampling, temperature, top_p, top_k, repetition_penalty, max_tokens], outputs=[output], queue=False)
     maintain.click(cloud, inputs=[], outputs=[], queue=False)
 
 main.launch(show_api=True)