Update app.py

app.py

@@ -13,6 +13,10 @@ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
 )
 processor = AutoProcessor.from_pretrained("daniel3303/QwenStoryteller")
 
+def upload_file(files):
+    file_paths = [file.name for file in files]
+    return file_paths
+    
 @spaces.GPU()
 def generate_story(images):
     image_content = []
@@ -68,12 +72,20 @@ def generate_story(images):
 
     return story
 
-demo = gr.Interface(
-    fn=generate_story,
-    inputs=gr.Image(type="pil", label="Upload up to 6 images", image_mode="RGB", height=300, width=300, file_types=[".jpg", ".jpeg", ".png", ".webp"]),
-    outputs=gr.Textbox(label="Generated Story", lines=10),
-    title="Qwen Storyteller",
-    description="Upload up to 6 images to generate a creative story."
+with gr.Blocks() as demo:
+    gr.Markdown("# Qwen Storyteller \n## Upload up to 6 images to generate a creative story.")
+    with gr.Row():
+        with gr.Column():
+            file_output = gr.File()
+            upload_button = gr.UploadButton("Upload up to 6 images", file_types=["image", "video"], file_count="multiple")
+            gen_button = gr.Button("Generate", variant="secondary")
+
+        with gr.Column():
+            outputs=gr.Textbox(label="Generated Story", lines=10)
+
+    upload_button.upload(upload_file, upload_button, file_output)
+    gen_button.click(fn=generate_story, file_output, outputs)
+    
 )
 
 if __name__ == "__main__":