update

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from PIL import Image
 
 import torch
 from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
@@ -18,5 +19,16 @@ pipe = StableDiffusionInpaintPipeline.from_pretrained(
 ).to(device)
 
 
+st.title("Stable Edit - Edit your photos with Stable Diffusion!")
+
+# upload image
+filename = st.file_uploader("upload an image")
+image = Image.open(filename)
+st.image(image)
+
+# Select Area to edit
+st.selectbox("Options", ("AutoSegment Area", "Draw Custom Area"))
+
+# TEMP - DEMO stuff
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)