Tharuneshwar commited on
Commit
e2bcb00
·
1 Parent(s): 42decf6

Model changed

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. gradio.ipynb +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from visualize.visualize import removeBgFromSegmentImage, removeOnlyBg
9
  from models.model import getMask, loadModel
10
  from models.preprocess import preprocess
11
 
12
- FAST_SAM = loadModel()
13
 
14
  # Main processing function
15
 
 
9
  from models.model import getMask, loadModel
10
  from models.preprocess import preprocess
11
 
12
+ FAST_SAM = loadModel('FastSAM.pt')
13
 
14
  # Main processing function
15
 
gradio.ipynb CHANGED
@@ -119,7 +119,7 @@
119
  },
120
  {
121
  "cell_type": "code",
122
- "execution_count": null,
123
  "metadata": {},
124
  "outputs": [
125
  {
 
119
  },
120
  {
121
  "cell_type": "code",
122
+ "execution_count": 7,
123
  "metadata": {},
124
  "outputs": [
125
  {