juxuan27 commited on
Commit
e5d0541
1 Parent(s): eccea34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -12,7 +12,10 @@ import random
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()
 
12
 
13
  import spaces
14
 
15
+ print("Installing correct gradio version...")
16
+ os.system("pip uninstall -y gradio")
17
  os.system("pip install gradio==3.50.0")
18
+ print("Installing Finished!")
19
 
20
  mobile_sam = sam_model_registry['vit_h'](checkpoint='data/ckpt/sam_vit_h_4b8939.pth')
21
  mobile_sam.eval()