kishkath commited on
Commit
1f9d444
·
1 Parent(s): b5e25d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ import matplotlib.pyplot as plt
12
  from ultralytics import FastSAM
13
  from ultralytics.models.fastsam import FastSAMPrompt
14
 
15
- def inference(image):
16
 
17
  #model = FastSAM('FastSAM.pt')
18
  model = FastSAM(params['./FastSAM.pt'])
 
12
  from ultralytics import FastSAM
13
  from ultralytics.models.fastsam import FastSAMPrompt
14
 
15
+ def inference(input_image,show_all=True,text_prompt=''):
16
 
17
  #model = FastSAM('FastSAM.pt')
18
  model = FastSAM(params['./FastSAM.pt'])