Spaces:
Build error
Build error
encounter1997
commited on
Commit
·
5d906f6
1
Parent(s):
248c435
use hf model
Browse files- gradio_demo/app_running.py +6 -6
- gradio_demo/runner.py +1 -2
gradio_demo/app_running.py
CHANGED
|
@@ -84,7 +84,7 @@ def create_demo(runner: Runner,
|
|
| 84 |
examples = [
|
| 85 |
[
|
| 86 |
'CompVis/stable-diffusion-v1-4',
|
| 87 |
-
"
|
| 88 |
'A car is moving on the road.',
|
| 89 |
8, 0, 1,
|
| 90 |
'A jeep car is moving on the desert.',
|
|
@@ -94,7 +94,7 @@ def create_demo(runner: Runner,
|
|
| 94 |
|
| 95 |
[
|
| 96 |
'CompVis/stable-diffusion-v1-4',
|
| 97 |
-
"
|
| 98 |
'A blackswan is swimming on the water.',
|
| 99 |
8, 0, 4,
|
| 100 |
'A white swan is swimming on the water.',
|
|
@@ -104,7 +104,7 @@ def create_demo(runner: Runner,
|
|
| 104 |
|
| 105 |
[
|
| 106 |
'CompVis/stable-diffusion-v1-4',
|
| 107 |
-
"
|
| 108 |
'A child is riding a bike on the road.',
|
| 109 |
8, 0, 1,
|
| 110 |
'A lego child is riding a bike on the road.',
|
|
@@ -114,7 +114,7 @@ def create_demo(runner: Runner,
|
|
| 114 |
|
| 115 |
[
|
| 116 |
'CompVis/stable-diffusion-v1-4',
|
| 117 |
-
"
|
| 118 |
'A jeep car is moving on the road.',
|
| 119 |
8, 0, 6,
|
| 120 |
'A jeep car is moving on the snow.',
|
|
@@ -124,7 +124,7 @@ def create_demo(runner: Runner,
|
|
| 124 |
|
| 125 |
[
|
| 126 |
'CompVis/stable-diffusion-v1-4',
|
| 127 |
-
"
|
| 128 |
'A rabbit is eating a watermelon.',
|
| 129 |
8, 0, 6,
|
| 130 |
'A puppy is eating an orange.',
|
|
@@ -134,7 +134,7 @@ def create_demo(runner: Runner,
|
|
| 134 |
|
| 135 |
[
|
| 136 |
'CompVis/stable-diffusion-v1-4',
|
| 137 |
-
"
|
| 138 |
'A brown bear is sitting on the ground.',
|
| 139 |
8, 0, 6,
|
| 140 |
'A black bear is sitting on the grass.',
|
|
|
|
| 84 |
examples = [
|
| 85 |
[
|
| 86 |
'CompVis/stable-diffusion-v1-4',
|
| 87 |
+
"data/car-moving.mp4",
|
| 88 |
'A car is moving on the road.',
|
| 89 |
8, 0, 1,
|
| 90 |
'A jeep car is moving on the desert.',
|
|
|
|
| 94 |
|
| 95 |
[
|
| 96 |
'CompVis/stable-diffusion-v1-4',
|
| 97 |
+
"data/black-swan.mp4",
|
| 98 |
'A blackswan is swimming on the water.',
|
| 99 |
8, 0, 4,
|
| 100 |
'A white swan is swimming on the water.',
|
|
|
|
| 104 |
|
| 105 |
[
|
| 106 |
'CompVis/stable-diffusion-v1-4',
|
| 107 |
+
"data/child-riding.mp4",
|
| 108 |
'A child is riding a bike on the road.',
|
| 109 |
8, 0, 1,
|
| 110 |
'A lego child is riding a bike on the road.',
|
|
|
|
| 114 |
|
| 115 |
[
|
| 116 |
'CompVis/stable-diffusion-v1-4',
|
| 117 |
+
"data/car-turn.mp4",
|
| 118 |
'A jeep car is moving on the road.',
|
| 119 |
8, 0, 6,
|
| 120 |
'A jeep car is moving on the snow.',
|
|
|
|
| 124 |
|
| 125 |
[
|
| 126 |
'CompVis/stable-diffusion-v1-4',
|
| 127 |
+
"data/rabbit-watermelon.mp4",
|
| 128 |
'A rabbit is eating a watermelon.',
|
| 129 |
8, 0, 6,
|
| 130 |
'A puppy is eating an orange.',
|
|
|
|
| 134 |
|
| 135 |
[
|
| 136 |
'CompVis/stable-diffusion-v1-4',
|
| 137 |
+
"data/brown-bear.mp4",
|
| 138 |
'A brown bear is sitting on the ground.',
|
| 139 |
8, 0, 6,
|
| 140 |
'A black bear is sitting on the grass.',
|
gradio_demo/runner.py
CHANGED
|
@@ -88,8 +88,7 @@ class Runner:
|
|
| 88 |
output_dir.mkdir(parents=True, exist_ok=True)
|
| 89 |
|
| 90 |
config = OmegaConf.load('configs/black-swan.yaml')
|
| 91 |
-
|
| 92 |
-
config.pretrained_model_path = "checkpoints/stable-diffusion-v1-4" # TODO
|
| 93 |
|
| 94 |
config.output_dir = output_dir.as_posix()
|
| 95 |
config.input_data.video_path = input_video.name # type: ignore
|
|
|
|
| 88 |
output_dir.mkdir(parents=True, exist_ok=True)
|
| 89 |
|
| 90 |
config = OmegaConf.load('configs/black-swan.yaml')
|
| 91 |
+
config.pretrained_model_path = self.download_base_model(model_path, token=input_token)
|
|
|
|
| 92 |
|
| 93 |
config.output_dir = output_dir.as_posix()
|
| 94 |
config.input_data.video_path = input_video.name # type: ignore
|