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 |
|