Upload folder using huggingface_hub

app.py

@@ -25,22 +25,21 @@ style_files = os.listdir('./style_images')
 style_options = {' '.join(style_file.split('.')[0].split('_')): f'./style_images/{style_file}' for style_file in style_files}
 
 @spaces.GPU(duration=35)
-def inference(content_image, style_image, style_strength, output_quality, progress=gr.Progress(track_tqdm=True)):
+def inference(content_image, style_image, style_strength, progress=gr.Progress(track_tqdm=True)):
     yield None
     print('-'*15)
     print('DATETIME:', datetime.datetime.now())
     print('STYLE:', style_image)
-    img_size = 1024 if output_quality else 512
+    img_size = 512
     content_img, original_size = load_img(content_image, img_size)
     content_img = content_img.to(device)
     style_img = load_img_from_path(style_options[style_image], img_size)[0].to(device)
     
     print('CONTENT IMG SIZE:', original_size)
     print('STYLE STRENGTH:', style_strength)
-    print('HIGH QUALITY:', output_quality)
 
     iters = style_strength
-    lr = 1e-1
+    lr = 5e-2
     alpha = 1
     beta = 1
 
@@ -95,16 +94,14 @@ with gr.Blocks(css=css) as demo:
         style_dropdown = gr.Radio(choices=list(style_options.keys()), label='Choose a style', value='Starry Night', type='value')
         with gr.Accordion('Adjustments', open=False):
             with gr.Group():
-                style_strength_slider = gr.Slider(label='Style Strength', minimum=0, maximum=100, step=5, value=50)
+                style_strength_slider = gr.Slider(label='Style Strength', minimum=1, maximum=100, step=1, value=50)
                 with gr.Row():
                     low_button = gr.Button('Low').click(fn=lambda: set_slider(10), outputs=[style_strength_slider])
                     medium_button = gr.Button('Medium').click(fn=lambda: set_slider(50), outputs=[style_strength_slider])
                     high_button = gr.Button('High').click(fn=lambda: set_slider(100), outputs=[style_strength_slider])
-            with gr.Group():
-                output_quality = gr.Checkbox(label='High Quality', info='Note: This takes longer, but improves output image quality')
         submit_button = gr.Button('Submit')
     
-        submit_button.click(fn=inference, inputs=[content_and_output, style_dropdown, style_strength_slider, output_quality], outputs=[content_and_output])
+        submit_button.click(fn=inference, inputs=[content_and_output, style_dropdown, style_strength_slider], outputs=[content_and_output])
         
         examples = gr.Examples(
             examples=[
@@ -112,7 +109,7 @@ with gr.Blocks(css=css) as demo:
                 ['./content_images/GoldenRetriever.jpg', 'Lego Bricks', 50, False],
                 ['./content_images/SeaTurtle.jpg', 'Mosaic', 100, False]
             ],
-            inputs=[content_and_output, style_dropdown, style_strength_slider, output_quality]
+            inputs=[content_and_output, style_dropdown, style_strength_slider]
         )
 
 # disable queue

vgg19.py

@@ -4,7 +4,7 @@ import torchvision.models as models
 class VGG_19(nn.Module):
     def __init__(self):
         super(VGG_19, self).__init__()
-        self.model = models.vgg19(pretrained=True).features[:30]
+        self.model = models.vgg19(weights='DEFAULT').features[:30]
         
         for i, _ in enumerate(self.model):
             if i in [4, 9, 18, 27]: