Mairaaa commited on
Commit
290f7fe
·
verified ·
1 Parent(s): c66a75c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ uploaded_file = st.file_uploader("Upload your rough sketch (PNG, JPG, JPEG):", t
12
 
13
  # Sidebar for Parameters
14
  st.sidebar.title("Model Configuration")
15
- pretrained_model_path = st.sidebar.text_input("Pretrained Model Path", "./models")
16
  dataset_path = st.sidebar.text_input("Dataset Path", "./datasets/dresscode")
17
  output_dir = st.sidebar.text_input("Output Directory", "./outputs")
18
  guidance_scale_sketch = st.sidebar.slider("Sketch Guidance Scale", 1.0, 10.0, 7.5)
 
12
 
13
  # Sidebar for Parameters
14
  st.sidebar.title("Model Configuration")
15
+ pretrained_model_path = st.sidebar.text_input("Pretrained Model Path", "runwayml/stable-diffusion-inpainting")
16
  dataset_path = st.sidebar.text_input("Dataset Path", "./datasets/dresscode")
17
  output_dir = st.sidebar.text_input("Output Directory", "./outputs")
18
  guidance_scale_sketch = st.sidebar.slider("Sketch Guidance Scale", 1.0, 10.0, 7.5)