Soooma commited on
Commit
5673cfe
·
verified ·
1 Parent(s): aafb47a

Update OK.jpg image path (as local file)

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ model = AutoModel.from_pretrained(model_name).to(device)
13
  processor = AutoImageProcessor.from_pretrained(model_name)
14
 
15
  # Load OK reference image (built-in normal)
16
- ok_image = Image.open("https://huggingface.co/datasets/Soooma/sofc/blob/main/OK1.jpg")
 
17
 
18
  # Precompute OK feature embedding
19
  with torch.no_grad():
 
13
  processor = AutoImageProcessor.from_pretrained(model_name)
14
 
15
  # Load OK reference image (built-in normal)
16
+ #ok_image = Image.open("https://huggingface.co/datasets/Soooma/sofc/blob/main/OK1.jpg")
17
+ ok_image = Image.open("OK1.jpg")
18
 
19
  # Precompute OK feature embedding
20
  with torch.no_grad():