Changed the initial values

.gitignore

@@ -0,0 +1 @@
+__pycache__

app.py

@@ -30,15 +30,15 @@ def change_dtype_inputs(
 def fit_style_transfer(
     style_image,
     content_image,
-    extractor="inception_v3",
-    n_style_layers=2,
-    n_content_layers=3,
-    epochs=4,
-    learning_rate=60.0,
-    steps_per_epoch=100,
-    style_weight=0.3,
-    content_weight=0.5,
-    var_weight=1e-12,
+    extractor,
+    n_style_layers,
+    n_content_layers,
+    epochs,
+    learning_rate,
+    steps_per_epoch,
+    style_weight,
+    content_weight,
+    var_weight,
 ):
     """
     Fit the style transfer model to the content and style images.
@@ -127,7 +127,7 @@ def fit_style_transfer(
 
     optimizer = tf.optimizers.Adam(
         tf.keras.optimizers.schedules.ExponentialDecay(
-            initial_learning_rate=learning_rate, decay_steps=100, decay_rate=0.80
+            initial_learning_rate=learning_rate, decay_steps=100, decay_rate=0.50
         )
     )
 
@@ -187,37 +187,37 @@ def main():
 
     epochs = gr.Slider(2, 20, value=4, step=1, label="Epochs")
 
-    learning_rate = gr.Slider(1, 100, value=60, step=1, label="Learning Rate")
+    learning_rate = gr.Slider(1, 50, value=20, step=1, label="Learning Rate")
 
     steps_per_epoch = gr.Slider(
         1,
-        100,
-        value=80,
+        10,
+        value=5,
         step=1,
         label="Steps Per Epoch",
     )
 
     style_weight = gr.Slider(
         1e-4,
-        0.5,
-        value=0.3,
+        1e-1,
+        value=5e-2,
         step=1e-4,
         label="Style Weight",
     )
 
     content_weight = gr.Slider(
         1e-3,
-        0.5,
-        value=0.5,
+        1e-1,
+        value=3e-2,
         step=1e-4,
         label="Content Weight",
     )
 
     var_weight = gr.Slider(
         0,
-        1e-5,
-        value=1e-7,
-        step=1e-12,
+        3,
+        value=1.5,
+        step=0.2,
         label="Total Variation Weight",
     )
 
@@ -241,13 +241,13 @@ def main():
             "examples/content_1.jpg",
             "inception_v3",
             3,
-            2,
-            4,
-            60,
-            100,
-            0.3,
-            0.5,
-            1e-8,
+            1,
+            5,
+            80,
+            10,
+            2e-2,
+            1e-2,
+            1.5,
         ],
         [
             "examples/style_2.jpg",
@@ -257,23 +257,24 @@ def main():
             2,
             4,
             60,
-            100,
-            0.3,
-            0.5,
-            1e-5,
+            10,
+            5e-2,
+            3e-2,
+            1.5,
         ],
+        
         [
             "examples/style_3.jpg",
             "examples/content_3.jpg",
             "inception_v3",
             3,
-            2,
+            1,
             4,
             60,
-            100,
-            0.5,
-            0.3,
-            1e-10,
+            10,
+            5e-2,
+            3e-2,
+            1.5,
         ]
 
     ]