ziyangmai commited on
Commit
dce64a0
·
1 Parent(s): 1ad8288
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -1,10 +1,5 @@
1
  import gradio as gr
2
  import os
3
- import torch
4
- import tempfile
5
- import random
6
- import string
7
- import json
8
  from omegaconf import OmegaConf,ListConfig
9
  import spaces
10
 
@@ -22,7 +17,15 @@ def inference_app(
22
  seed,
23
  inference_steps):
24
 
25
- return inference_main(embedding_dir, prompt, video_round, save_dir, motion_type, seed, inference_steps)
 
 
 
 
 
 
 
 
26
 
27
 
28
  def train_model(video, config):
 
1
  import gradio as gr
2
  import os
 
 
 
 
 
3
  from omegaconf import OmegaConf,ListConfig
4
  import spaces
5
 
 
17
  seed,
18
  inference_steps):
19
 
20
+ return inference_main(
21
+ embedding_dir=embedding_dir,
22
+ prompt=prompt,
23
+ video_round=video_round,
24
+ save_dir=save_dir,
25
+ motion_type=motion_type,
26
+ seed=seed,
27
+ inference_steps=inference_steps
28
+ )
29
 
30
 
31
  def train_model(video, config):