bedead commited on
Commit
9b720eb
·
1 Parent(s): 7d9a8b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(share=True)
 
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()