Update app.py

app.py

@@ -6,8 +6,8 @@ from datasets import Dataset
 from torch.nn import CosineSimilarity
 
 image_processor = ViTImageProcessor.from_pretrained("vit-base-patch16-224")
-image_encoder = ViTModel.from_pretrained("output/image_encoder/epoch_29").eval().to("cuda")
-scribble_encoder = ViTModel.from_pretrained("output/scibble_encoder/epoch_29").eval().to("cuda")
+image_encoder = ViTModel.from_pretrained("model/image_encoder/epoch_29").eval().to("cuda")
+scribble_encoder = ViTModel.from_pretrained("model/scibble_encoder/epoch_29").eval().to("cuda")
 
 candidates: Dataset = None