juxuan27 commited on
Commit
eccea34
·
verified ·
1 Parent(s): b32a57f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -12,6 +12,8 @@ import random
12
 
13
  import spaces
14
 
 
 
15
  mobile_sam = sam_model_registry['vit_h'](checkpoint='data/ckpt/sam_vit_h_4b8939.pth')
16
  mobile_sam.eval()
17
  mobile_predictor = SamPredictor(mobile_sam)
 
12
 
13
  import spaces
14
 
15
+ os.system("pip install gradio==3.50.0")
16
+
17
  mobile_sam = sam_model_registry['vit_h'](checkpoint='data/ckpt/sam_vit_h_4b8939.pth')
18
  mobile_sam.eval()
19
  mobile_predictor = SamPredictor(mobile_sam)