Update app.py

app.py

@@ -68,7 +68,7 @@ class resnet_feature_extractor(torch.nn.Module):
 
         return patch
 
-image = Image.open(r'/content/transistor/test/good/000.png')
+image = Image.open(r'transistor/test/good/000.png')
 image = transform(image).unsqueeze(0)
 
 backbone = resnet_feature_extractor()
@@ -83,7 +83,7 @@ print(feature.shape)
 
 memory_bank =[]
 
-folder_path = Path(r'/content/transistor/train/good')
+folder_path = Path(r'transistor/train/good')
 
 for pth in tqdm(folder_path.iterdir(),leave=False):
     with torch.no_grad():
@@ -95,7 +95,7 @@ memory_bank = torch.cat(memory_bank,dim=0)
 
 y_score=[]
 
-folder_path = Path(r'/content/transistor/train/good')
+folder_path = Path(r'transistor/train/good')
 
 for pth in tqdm(folder_path.iterdir(),leave=False):
     data = transform(Image.open(pth)).unsqueeze(0)
@@ -121,7 +121,7 @@ y_score = []
 y_true=[]
 
 for classes in ['bent_lead','good','cut_lead','damaged_case','misplaced']:
-    folder_path = Path(r'/content/transistor/test/{}'.format(classes))
+    folder_path = Path(r'transistor/test/{}'.format(classes))
 
     for pth in tqdm(folder_path.iterdir(),leave=False):
 
@@ -146,7 +146,7 @@ plt.vlines(x=best_threshold,ymin=0,ymax=30,color='r')
 plt.show()
 
 
-test_image = transform(Image.open(r'/content/transistor/test/good/000.png')).unsqueeze(0)
+test_image = transform(Image.open(r'transistor/test/good/000.png')).unsqueeze(0)
 features = backbone(test_image)
 
 distances = torch.cdist(features, memory_bank, p=2.0)