Update app.py

app.py

@@ -252,7 +252,7 @@ def start_items():
     thick = debias(thick, "Pale_Skin", df, pinverse, device)
     thick = debias(thick, "Heavy_Makeup", df, pinverse, device)
 
-
+start_items()
 
 class CustomImageDataset(Dataset):
     def __init__(self, images, transform=None):
@@ -417,7 +417,7 @@ with gr.Blocks(css="style.css") as demo:
                 with gr.Column(): 
                     gr.Markdown("""1) Either sample a new model, or upload an image (optionally draw a mask over the head) and click `invert`.""")
                     sample = gr.Button("🎲 Sample New Model")
-                    input_image = gr.Image(source='upload', elem_id="image_upload", tool='sketch', type='pil', label="Reference Identity",
+                    input_image = gr.Image(elem_id="image_upload", tool='sketch', type='pil', label="Reference Identity",
                                            width=512, height=512, brush_color='#00FFFF', mask_opacity=0.4)    
                   
                     with gr.Row():