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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,6 +1,5 @@
1
  ##!/usr/bin/python3
2
  # -*- coding: utf-8 -*-
3
- import gradio as gr
4
  import os
5
  import cv2
6
  from PIL import Image
@@ -17,6 +16,8 @@ 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()
22
  mobile_predictor = SamPredictor(mobile_sam)
 
1
  ##!/usr/bin/python3
2
  # -*- coding: utf-8 -*-
 
3
  import os
4
  import cv2
5
  from PIL import Image
 
16
  os.system("pip install gradio==3.50.0")
17
  print("Installing Finished!")
18
 
19
+ import gradio as gr
20
+
21
  mobile_sam = sam_model_registry['vit_h'](checkpoint='data/ckpt/sam_vit_h_4b8939.pth')
22
  mobile_sam.eval()
23
  mobile_predictor = SamPredictor(mobile_sam)