From 6cf5efc8b4a69ac0bd5322f4c48cbe2f407541a8 Mon Sep 17 00:00:00 2001
From: Akash A Desai <62583018+akashAD98@users.noreply.github.com>
Date: Mon, 26 Jun 2023 14:36:06 +0530
Subject: [PATCH 1/2] Update visualizer_drag_gradio.py
added support for google collab & gradio
,just replaced weights & update share='True' as default so people will use it
---
visualizer_drag_gradio.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/visualizer_drag_gradio.py b/visualizer_drag_gradio.py
index 699787d..42e6c15 100644
--- a/visualizer_drag_gradio.py
+++ b/visualizer_drag_gradio.py
@@ -15,7 +15,7 @@ from gradio_utils import (ImageMask, draw_mask_on_image, draw_points_on_image,
from viz.renderer import Renderer, add_watermark_np
parser = ArgumentParser()
-parser.add_argument('--share', action='store_true')
+parser.add_argument('--share', action='store_true',default='True')
parser.add_argument('--cache-dir', type=str, default='./checkpoints')
args = parser.parse_args()
@@ -156,7 +156,7 @@ print(os.listdir(cache_dir))
print('Valid checkpoint file:')
print(valid_checkpoints_dict)
-init_pkl = 'stylegan_human_v2_512'
+init_pkl = 'stylegan2_lions_512_pytorch'
with gr.Blocks() as app:
From ee253bac086022eaf97e61136bb91ff7a4dd37da Mon Sep 17 00:00:00 2001
From: Akash A Desai <62583018+akashAD98@users.noreply.github.com>
Date: Mon, 26 Jun 2023 14:36:56 +0530
Subject: [PATCH 2/2] Update README.md
---
README.md | 1 +
1 file changed, 1 insertion(+)
diff --git a/README.md b/README.md
index 1396a99..5593ee4 100644
--- a/README.md
+++ b/README.md
@@ -29,6 +29,7 @@
+ 