Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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,
|