update

app.py

@@ -1,11 +1,16 @@
 import os
+os.system('wget https://anaconda.org/pytorch3d/pytorch3d/0.7.7/download/linux-64/pytorch3d-0.7.7-py39_cu117_pyt200.tar.bz2')
+os.system('conda install pytorch3d-0.7.7-py39_cu117_pyt200.tar.bz2')
+os.system('rm -r pytorch3d-0.7.7-py39_cu117_pyt200.tar.bz2')
+os.system('wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/')
+
 import sys
 import gradio as gr
 import random
 from viewcrafter import ViewCrafter
 from configs.infer_config import get_parser
 import torch
-
+from huggingface_hub import hf_hub_download
 
 i2v_examples = [
     ['test/images/boy.png', 0, 1.0, '0 40', '0 0', '0 0',  50, 123],
@@ -17,9 +22,17 @@ i2v_examples = [
 
 max_seed = 2 ** 31
 
-os.system('conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.7/download/linux-64/pytorch3d-0.7.7-py39_cu117_pyt200.tar.bz2')
+def download_model():
+    REPO_ID = 'Drexubery/ViewCrafter_25'
+    filename_list = ['model.ckpt']
+    for filename in filename_list:
+        local_file = os.path.join('./checkpoints/', filename)
+        if not os.path.exists(local_file):
+            hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./checkpoints/', force_download=True)
+
 
 
+download_model()
 
 def viewcrafter_demo(opts):
     css = """#input_img {max-width: 1024px !important} #output_vid {max-width: 1024px; max-height:576px} #random_button {max-width: 100px !important}"""

requirements.txt

@@ -38,4 +38,5 @@ Pygments==2.18.0
 starlette==0.38.2
 tifffile==2024.8.24
 yarl==1.9.4
-spaces
+spaces
+huggingface_hub