Daniel-F commited on
Commit
4c6b11a
·
1 Parent(s): 187d444
Files changed (2) hide show
  1. app.py +1 -0
  2. install.sh +3 -3
app.py CHANGED
@@ -126,6 +126,7 @@ with gr.Blocks() as demo:
126
  output_mask = gr.Image(type="pil", label="Generated Mask")
127
 
128
  reference_img.select(fn=record_click, inputs=[reference_img], outputs=[click_info])
 
129
  generate_btn.click(fn=generate, inputs=[reference_img, target_img], outputs=[output_mask, click_info])
130
 
131
  global sam2_img
 
126
  output_mask = gr.Image(type="pil", label="Generated Mask")
127
 
128
  reference_img.select(fn=record_click, inputs=[reference_img], outputs=[click_info])
129
+ reference_img.change(fn=on_reference_upload, inputs=[reference_img], outputs=[click_info])
130
  generate_btn.click(fn=generate, inputs=[reference_img, target_img], outputs=[output_mask, click_info])
131
 
132
  global sam2_img
install.sh CHANGED
@@ -1,8 +1,8 @@
1
  #!/bin/bash
2
 
3
- # conda create -y --name sst python=3.10.14
4
- # source ~/anaconda3/etc/profile.d/conda.sh
5
- # conda activate sst
6
  pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
7
  (cd segment-anything-2 && pip install -e .)
8
  pip install -r requirements_manual.txt --no-dependencies
 
1
  #!/bin/bash
2
 
3
+ conda create -y --name sst python=3.10.14
4
+ source ~/anaconda3/etc/profile.d/conda.sh
5
+ conda activate sst
6
  pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
7
  (cd segment-anything-2 && pip install -e .)
8
  pip install -r requirements_manual.txt --no-dependencies