Spaces:
Sleeping
Sleeping
hengli
commited on
Commit
·
424fc2a
1
Parent(s):
acbafbf
update hf link
Browse files
app.py
CHANGED
|
@@ -38,8 +38,8 @@ else:
|
|
| 38 |
model.load_state_dict(torch.hub.load_state_dict_from_url(_URL))
|
| 39 |
|
| 40 |
|
| 41 |
-
model.eval()
|
| 42 |
-
model = model.to(device)
|
| 43 |
|
| 44 |
|
| 45 |
# -------------------------------------------------------------------------
|
|
@@ -58,7 +58,7 @@ def run_model(target_dir, model, anchor_size=100) -> dict:
|
|
| 58 |
raise ValueError("CUDA is not available. Check your environment.")
|
| 59 |
|
| 60 |
# Move model to device
|
| 61 |
-
|
| 62 |
model.eval()
|
| 63 |
|
| 64 |
# Load and preprocess images
|
|
|
|
| 38 |
model.load_state_dict(torch.hub.load_state_dict_from_url(_URL))
|
| 39 |
|
| 40 |
|
| 41 |
+
# model.eval()
|
| 42 |
+
# model = model.to(device)
|
| 43 |
|
| 44 |
|
| 45 |
# -------------------------------------------------------------------------
|
|
|
|
| 58 |
raise ValueError("CUDA is not available. Check your environment.")
|
| 59 |
|
| 60 |
# Move model to device
|
| 61 |
+
model = model.to(device)
|
| 62 |
model.eval()
|
| 63 |
|
| 64 |
# Load and preprocess images
|