Added configuration for Streamlit server in config.toml

.streamlit/config.toml

@@ -0,0 +1,4 @@
+[server]
+enableXsrfProtection = false
+enableCORS = false
+maxUploadSize = 10

src/app.py

@@ -9,7 +9,6 @@ from torchvision import transforms as tr
 import io
 
 
-# ===== Model definitions =====
 class ResidualBlock(nn.Module):
     def __init__(self, channels):
         super(ResidualBlock, self).__init__()
@@ -111,7 +110,6 @@ class CycleGAN(nn.Module):
         })
 
 
-# ===== Load model =====
 @st.cache_resource
 def load_model():
     model_dir = os.path.dirname(os.path.abspath(__file__))
@@ -143,7 +141,6 @@ def postprocess(tensor, mean, std):
     return Image.fromarray(img)
 
 
-# ===== Streamlit UI =====
 st.set_page_config(page_title="CycleGAN: Summer <-> Winter", layout="wide")
 st.title("🏔️ CycleGAN: Summer ↔ Winter (Yosemite)")
 st.markdown("Upload an image and transform it between summer and winter styles!")