刘虹雨 commited on
Commit
409cac4
·
1 Parent(s): 0a9d3cd

update code

Browse files
Files changed (3) hide show
  1. .DS_Store +0 -0
  2. app.py +5 -4
  3. demo_data/.DS_Store +0 -0
.DS_Store CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
 
app.py CHANGED
@@ -358,7 +358,8 @@ def launch_pretrained():
358
  )
359
  logging.info("delete models.")
360
 
361
-
 
362
 
363
  # 下载 CrucibleAI/ControlNetMediaPipeFace 的所有文件
364
  snapshot_download(
@@ -860,7 +861,7 @@ def launch_gradio_app():
860
  with gr.TabItem('🎨 Upload Image'):
861
  input_image = gr.Image(
862
  label="Upload Source Image",
863
- value=os.path.join(image_folder, '02057_2.png'),
864
  image_mode="RGB", height=512, container=True,
865
  sources="upload", type="filepath"
866
  )
@@ -902,7 +903,7 @@ def launch_gradio_app():
902
  video_input = gr.Video(
903
  label="Select Target Motion",
904
  height=512, container=True,interactive=False, format="mp4",
905
- value=examples_videos[0]
906
  )
907
 
908
  with gr.Row():
@@ -942,7 +943,7 @@ def launch_gradio_app():
942
  label="CFG Scale"
943
  )
944
  strength_slider = gr.Slider(
945
- minimum=0.4, maximum=0.85, value=0.75, step=0.05,
946
  label="SDEdit Strength"
947
  )
948
  style_button = gr.Button("🎨 Apply Style", interactive=False)
 
358
  )
359
  logging.info("delete models.")
360
 
361
+ # os.remove('./pretrained_model/sd21/v2-1_512-ema-pruned.ckpt')
362
+ # os.remove('./pretrained_model/sd21/v2-1_512-nonema-pruned.ckpt')
363
 
364
  # 下载 CrucibleAI/ControlNetMediaPipeFace 的所有文件
365
  snapshot_download(
 
861
  with gr.TabItem('🎨 Upload Image'):
862
  input_image = gr.Image(
863
  label="Upload Source Image",
864
+ value=os.path.join(image_folder, '02025.png'),
865
  image_mode="RGB", height=512, container=True,
866
  sources="upload", type="filepath"
867
  )
 
903
  video_input = gr.Video(
904
  label="Select Target Motion",
905
  height=512, container=True,interactive=False, format="mp4",
906
+ value=os.path.join(video_folder, 'Obama.mp4')
907
  )
908
 
909
  with gr.Row():
 
943
  label="CFG Scale"
944
  )
945
  strength_slider = gr.Slider(
946
+ minimum=0.5, maximum=0.85, value=0.75, step=0.05,
947
  label="SDEdit Strength"
948
  )
949
  style_button = gr.Button("🎨 Apply Style", interactive=False)
demo_data/.DS_Store CHANGED
Binary files a/demo_data/.DS_Store and b/demo_data/.DS_Store differ