Update app.py

app.py

@@ -17,7 +17,7 @@ tf.config.threading.set_intra_op_parallelism_threads(8)
 
 def load_img(image):
     """Load and preprocess image for style transfer"""
-    max_dim = 256
+    max_dim = 1024
     # Convert PIL Image to tensor
     img = tf.convert_to_tensor(np.array(image))
     img = tf.image.convert_image_dtype(img, tf.float32)
@@ -129,10 +129,8 @@ def style_transfer_fn(content_image, style_image, progress=gr.Progress(track_tqd
         style_weight = 1e-2
         content_weight = 1e4
         total_variation_weight = 30
-
         epochs = 10
         steps_per_epoch = 100
-
         start_time = time.time()
 
         # Training loop
@@ -158,12 +156,8 @@ iface = gr.Interface(
         gr.Image(label="Style Image", type="pil")
     ],
     outputs=gr.Image(label="Stylized Image"),
-    title="Neural Style Transfer",
-    description="Upload a content image and a style image to create a stylized combination.",
-    examples=[
-        ["examples/content.jpg", "examples/style.jpg"]
-    ],
-    cache_examples=True
+    title="Neural Style Transfer - Ty Chermsirivatana",
+    description="Upload a content image and a style image to create a stylized image in context.",
 )
 
 # Launch the interface