Upload 2 files

app.py

@@ -43,6 +43,7 @@ def ui():
                 with gr.Column():
                     gr.Image(value=image, label=f"{label}",
                              height=IMAGE_SIZE,
+                             width=IMAGE_SIZE,
                              show_download_button=False,
                              show_share_button=False)
                     #gr.Textbox(value=f"Label {label}", interactive=False)

utils.py

@@ -10,14 +10,14 @@ import torch.nn.functional as F
 import random
 
 style_guide = [
-    ('Oil Style', 'oilstyle_learned_embeds.bin'),
+    ('Oil Painting Style', 'oilstyle_learned_embeds.bin'),
     ('Matrix Style', 'matrix_learned_embeds.bin'),
     ('Stripe Style', 'stripe_learned_embeds.bin'),
     ('Dreamy Painting Style', 'dreamypainting_learned_embeds.bin'),
     ('Polygon HD Style', 'lowpolyhd_learned_embeds.bin')
 ]
 
-IMAGE_SIZE = 224
+IMAGE_SIZE = 150
 
 # Prep Scheduler
 def set_timesteps(scheduler, num_inference_steps):