Skip to content

Commit

Permalink
[fix](gradio): download model firstly, then load model in import
Browse files Browse the repository at this point in the history
  • Loading branch information
itechmusic committed Mar 30, 2024
1 parent 5338d84 commit 55e1e28
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions scripts/gradio/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,7 @@
import pdb
import cuid

from gradio_text2video import online_t2v_inference
from gradio_video2video import online_v2v_inference

from huggingface_hub import snapshot_download

ProjectDir = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
Expand All @@ -25,6 +24,9 @@ def download_model():

download_model() # for huggingface deployment.

from gradio_text2video import online_t2v_inference
from gradio_video2video import online_v2v_inference


def update_shape(image):
if image != None:
Expand Down

0 comments on commit 55e1e28

Please sign in to comment.