Update OK.jpg image path (as local file)
Browse files
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():
|