Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ format = { 0: 'Adenocarcinoma case',
|
|
12 |
|
13 |
# returning classifiers output
|
14 |
def image_classifier(inp):
|
15 |
-
model = YOLO("best.pt")
|
16 |
|
17 |
result = model.predict(source=inp)
|
18 |
probs = result[0].probs
|
@@ -39,4 +39,4 @@ with gr.Blocks() as app:
|
|
39 |
cache_examples=True,
|
40 |
)
|
41 |
|
42 |
-
app.launch(
|
|
|
12 |
|
13 |
# returning classifiers output
|
14 |
def image_classifier(inp):
|
15 |
+
model = YOLO("best-2.pt")
|
16 |
|
17 |
result = model.predict(source=inp)
|
18 |
probs = result[0].probs
|
|
|
39 |
cache_examples=True,
|
40 |
)
|
41 |
|
42 |
+
app.launch()
|