Update app.py

app.py

@@ -82,5 +82,14 @@ def main(user_image):
   # sample the new 
   return sample_images(init_image, pil_inverted_mask)
 
-demo = gr.Interface(main, gr.File(), "image")
+with gr.Blocks() as demo:
+    gr.Markdown(
+    """
+    # Upload a selfie/image of yourself/your friends, and get a funny, professional headshot :)
+    """)
+    inp = gr.File()
+    out = gr.Image()
+    inp.change(main, inp, out)
+    
+#demo = gr.Interface(main, gr.File(), "image")
 demo.launch(debug=True)