Spaces:
Runtime error
Runtime error
debug app
Browse files
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(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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):
|