ahsanMah commited on
Commit
932b692
·
1 Parent(s): e71c8e0

hf app fix

Browse files
Files changed (1) hide show
  1. hfapp.py +2 -2
hfapp.py CHANGED
@@ -33,10 +33,10 @@ def localize_anomalies(input_img, preset="edm2-img64-s-fid", load_from_hub=False
33
  img = np.array(input_img)
34
  img = torch.from_numpy(img).permute(2, 0, 1).unsqueeze(0)
35
  img = img.float().to(device)
36
- model = load_model_from_hub(preset=preset, device=device)
37
  img_likelihood, score_norms = run_inference(model, img)
38
  nll, pct, ref_nll = compute_gmm_likelihood(
39
- score_norms, model_dir=f"models/{preset}"
40
  )
41
 
42
  outstr = f"Anomaly score: {nll:.3f} / {pct:.2f} percentile"
 
33
  img = np.array(input_img)
34
  img = torch.from_numpy(img).permute(2, 0, 1).unsqueeze(0)
35
  img = img.float().to(device)
36
+ model, modeldir = load_model_from_hub(preset=preset, device=device)
37
  img_likelihood, score_norms = run_inference(model, img)
38
  nll, pct, ref_nll = compute_gmm_likelihood(
39
+ score_norms, model_dir=modeldir
40
  )
41
 
42
  outstr = f"Anomaly score: {nll:.3f} / {pct:.2f} percentile"