Swekerr commited on
Commit
92eea01
1 Parent(s): b309cea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def classify_brain_tumor(image):
61
  with torch.no_grad():
62
  output = model(image)
63
  prediction = torch.sigmoid(output).item()
64
- return "Tumor" if prediction >= 0.75 else "No Tumor"
65
 
66
  interface = gr.Interface(
67
  fn=classify_brain_tumor,
 
61
  with torch.no_grad():
62
  output = model(image)
63
  prediction = torch.sigmoid(output).item()
64
+ return "Tumor" if prediction >= 0.5 else "No Tumor"
65
 
66
  interface = gr.Interface(
67
  fn=classify_brain_tumor,