mjohanes commited on
Commit
46db2a8
·
1 Parent(s): a4f6b0d
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ from diffusers import StableDiffusionInpaintPipeline
16
  from ultralytics import FastSAM
17
 
18
  # Set page config for a better mobile experience
19
- st.set_page_config(page_title="Inpainting Demo", layout="wide")
20
 
21
  # Define model paths or IDs for easy switching in the future
22
  FASTSAM_CHECKPOINT = "FastSAM-x.pt" # file name of the FastSAM model weights
@@ -40,7 +40,7 @@ if not os.path.exists(FASTSAM_CHECKPOINT):
40
  # Here we use the official Ultralytics release URL for FastSAM-x (68MB).
41
  import requests
42
  fastsam_url = "https://github.com/ultralytics/assets/releases/download/v8.2.0/FastSAM-x.pt"
43
- st.write("Downloading FastSAM model weights...")
44
  resp = requests.get(fastsam_url)
45
  open(FASTSAM_CHECKPOINT, "wb").write(resp.content)
46
 
 
16
  from ultralytics import FastSAM
17
 
18
  # Set page config for a better mobile experience
19
+ st.set_page_config(page_title="Inpainting Demo", layout="centered")
20
 
21
  # Define model paths or IDs for easy switching in the future
22
  FASTSAM_CHECKPOINT = "FastSAM-x.pt" # file name of the FastSAM model weights
 
40
  # Here we use the official Ultralytics release URL for FastSAM-x (68MB).
41
  import requests
42
  fastsam_url = "https://github.com/ultralytics/assets/releases/download/v8.2.0/FastSAM-x.pt"
43
+ # st.write("Downloading FastSAM model weights...")
44
  resp = requests.get(fastsam_url)
45
  open(FASTSAM_CHECKPOINT, "wb").write(resp.content)
46