Baptlem commited on
Commit
35f97ba
1 Parent(s): 207b3ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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="./trajectory/trajectory.avi",
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