Upload app.py

app.py

@@ -0,0 +1,57 @@
+
+import streamlit as st
+from PIL import Image
+import torch
+import torchvision.transforms as transforms
+from animegan2.model import Generator
+import os
+import urllib.request
+
+st.set_page_config(page_title="Ghibli Style Image Converter 🎨")
+
+MODEL_PATH = "weights/paprika.pt"
+MODEL_URL = "https://huggingface.co/akhaliq/AnimeGANv2/resolve/main/paprika.pt"
+
+@st.cache_resource
+def load_model():
+    if not os.path.exists(MODEL_PATH):
+        os.makedirs("weights", exist_ok=True)
+        with st.spinner("Downloading model weights..."):
+            urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
+            st.success("Model downloaded successfully!")
+    model = Generator()
+    model.load_state_dict(torch.load(MODEL_PATH, map_location="cpu"))
+    model.eval()
+    return model
+
+def preprocess(image):
+    image = image.convert("RGB")
+    transform = transforms.Compose([
+        transforms.Resize((256, 256)),
+        transforms.ToTensor()
+    ])
+    return transform(image).unsqueeze(0)
+
+def postprocess(tensor):
+    tensor = tensor.squeeze().detach().cpu().clamp(0, 1)
+    image = transforms.ToPILImage()(tensor)
+    return image
+
+st.title("🎨 Convert to Ghibli Style!")
+st.markdown("Upload an image to transform it into a beautiful Ghibli-style artwork using AnimeGAN2!")
+
+uploaded_file = st.file_uploader("Upload your image", type=["jpg", "jpeg", "png"])
+
+if uploaded_file:
+    input_image = Image.open(uploaded_file)
+    st.image(input_image, caption="Original Image", use_column_width=True)
+
+    with st.spinner("Converting... please wait!"):
+        model = load_model()
+        input_tensor = preprocess(input_image)
+        with torch.no_grad():
+            output_tensor = model(input_tensor)
+        output_image = postprocess(output_tensor)
+
+    st.image(output_image, caption="Ghibli Style Image", use_column_width=True)
+    st.success("Done!")