JahDaGanj commited on
Commit
231b2a9
·
verified ·
1 Parent(s): e8a458e

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import numpy as np
3
+ import cv2
4
+ from PIL import Image
5
+ import torch
6
+ import torchvision.transforms as T
7
+ from torchvision.models.segmentation import deeplabv3_resnet101
8
+ from io import BytesIO
9
+
10
+ st.set_page_config(page_title="Kleidung umfärben", layout="centered")
11
+ st.title("🧥 Kleidung umfärben – Color Changer")
12
+
13
+ uploaded_file = st.file_uploader("📤 Bild hochladen", type=["jpg", "jpeg", "png"])
14
+
15
+ @st.cache_resource
16
+ def load_model():
17
+ model = deeplabv3_resnet101(pretrained=True).eval()
18
+ return model
19
+
20
+ def segment_clothes(img_pil):
21
+ transform = T.Compose([