Update app.py

app.py

@@ -33,13 +33,15 @@ def generate(image, video, instruction=DEFAULT_INPUT, sampling=False, temperatur
 
     print(image)
     print(video)
-    print(not video)
     
-    image_data = Image.fromarray(image.astype('uint8'), 'RGB')
-    print(image_data, instruction)
-
-    inputs = [{"role": "user", "content": [image_data, instruction]}]
+    if not video:
+        image_data = Image.fromarray(image.astype('uint8'), 'RGB')
+        print(image_data, instruction)
 
+        inputs = [{"role": "user", "content": [image_data, instruction]}]
+    else:
+        priny("video")
+    
     parameters = {
         "sampling": sampling,
         "temperature": temperature,
@@ -50,7 +52,9 @@ def generate(image, video, instruction=DEFAULT_INPUT, sampling=False, temperatur
     }
     
     output = repo.chat(image=None, msgs=inputs, tokenizer=tokenizer, **parameters)
+    
     print(output)
+    
     return output
     
 def cloud():
@@ -63,7 +67,7 @@ with gr.Blocks(css=css) as main:
         
     with gr.Column():
         input = gr.Image(label="Image")
-        input_2 = gr.Image(label="Video")
+        input_2 = gr.Video(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")