Update

Dockerfile

@@ -43,6 +43,13 @@ RUN pyenv install ${PYTHON_VERSION} && \
 COPY --chown=1000 ./requirements.txt /tmp/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt
 
+RUN git clone https://github.com/modelscope/modelscope && \
+    cd modelscope && \
+    pip install -r requirements.txt && \
+    pip install . && \
+    cd .. && \
+    rm -rf modelscope
+
 COPY --chown=1000 . ${HOME}/app
 ENV PYTHONPATH=${HOME}/app \
     PYTHONUNBUFFERED=1 \

app.py

@@ -11,6 +11,8 @@ from modelscope.outputs import OutputKeys
 from modelscope.pipelines import pipeline
 
 DESCRIPTION = "# ModelScope-Image2Video"
+if not torch.cuda.is_available():
+    DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
 if torch.cuda.is_available():
     model_cache_dir = os.getenv("MODEL_CACHE_DIR", "./models")
@@ -29,9 +31,16 @@ def image_to_video(image_path: str) -> str:
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
-    input_image = gr.Image(label="Input image", type="filepath")
-    run_button = gr.Button()
-    output_video = gr.Video(label="Output video")
+    gr.DuplicateButton(
+        value="Duplicate Space for private use",
+        elem_id="duplicate-button",
+        visible=os.getenv("SHOW_DUPLICATE_BUTTON") == "1",
+    )
+    with gr.Group():
+        input_image = gr.Image(label="Input image", type="filepath")
+        run_button = gr.Button()
+        output_video = gr.Video(label="Output video")
+
     run_button.click(
         fn=image_to_video,
         inputs=input_image,
@@ -40,4 +49,4 @@ with gr.Blocks(css="style.css") as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    demo.queue(max_size=10).launch()

requirements.txt

@@ -1,7 +1,7 @@
 easydict==1.10
 einops==0.6.1
 fairscale==0.4.13
-gradio==3.41.1
+gradio==3.41.2
 huggingface_hub==0.16.4
 imageio==2.31.1
 modelscope==1.8.4