Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ import cv2
|
|
11 |
from PIL import Image
|
12 |
import numpy as np
|
13 |
import gradio as gr
|
|
|
14 |
|
15 |
description = """
|
16 |
Our project is to use diffusion model to change the texture of our robotic arm simulation.
|
@@ -73,7 +74,8 @@ controlnet_version = "coyo-500k"
|
|
73 |
low_threshold = 100
|
74 |
high_threshold = 200
|
75 |
|
76 |
-
|
|
|
77 |
|
78 |
# pipe.enable_xformers_memory_efficient_attention()
|
79 |
# pipe.enable_model_cpu_offload()
|
@@ -233,7 +235,7 @@ def create_demo(process, max_images=12, default_num_images=4):
|
|
233 |
with gr.Row():
|
234 |
gr.Markdown(description)
|
235 |
|
236 |
-
gr.Video(value="
|
237 |
format="avi",
|
238 |
interactive=False)
|
239 |
|
|
|
11 |
from PIL import Image
|
12 |
import numpy as np
|
13 |
import gradio as gr
|
14 |
+
import os
|
15 |
|
16 |
description = """
|
17 |
Our project is to use diffusion model to change the texture of our robotic arm simulation.
|
|
|
74 |
low_threshold = 100
|
75 |
high_threshold = 200
|
76 |
|
77 |
+
print(os.path.abspath('.'))
|
78 |
+
print(os.listdir("."))
|
79 |
|
80 |
# pipe.enable_xformers_memory_efficient_attention()
|
81 |
# pipe.enable_model_cpu_offload()
|
|
|
235 |
with gr.Row():
|
236 |
gr.Markdown(description)
|
237 |
|
238 |
+
gr.Video(value="Baptlem/controlnet-for-data-augmentation/trajectory/trajectory.avi",
|
239 |
format="avi",
|
240 |
interactive=False)
|
241 |
|