Update app.py
Browse files
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(
|
| 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'])
|