hengli commited on
Commit
424fc2a
·
1 Parent(s): acbafbf

update hf link

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- # model = model.to(device)
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