Spaces:
Runtime error
Runtime error
Orpheous1
commited on
Commit
•
0422277
1
Parent(s):
523f190
no_grad
Browse files
app.py
CHANGED
|
@@ -48,6 +48,7 @@ def draw_heatmap(image, mask):
|
|
| 48 |
|
| 49 |
|
| 50 |
# Define callable method for the demo
|
|
|
|
| 51 |
def get_mask(image, model_name: str):
|
| 52 |
torch.manual_seed(seed=0)
|
| 53 |
if image is None:
|
|
|
|
| 48 |
|
| 49 |
|
| 50 |
# Define callable method for the demo
|
| 51 |
+
@torch.no_grad()
|
| 52 |
def get_mask(image, model_name: str):
|
| 53 |
torch.manual_seed(seed=0)
|
| 54 |
if image is None:
|