fixing file path in app.py
Browse files
app.py
CHANGED
|
@@ -18,12 +18,12 @@ labels = ['Defect', 'Non-Defect']
|
|
| 18 |
|
| 19 |
# Define the sample images
|
| 20 |
sample_images = {
|
| 21 |
-
"Defect01": "
|
| 22 |
-
"Defect02": "
|
| 23 |
-
"Defect03": "
|
| 24 |
-
"Non-Defect01": "
|
| 25 |
-
"Non-Defect02": "
|
| 26 |
-
"Non-Defect03": "
|
| 27 |
}
|
| 28 |
|
| 29 |
class DefectResNet(pl.LightningModule):
|
|
@@ -99,7 +99,7 @@ class DefectResNet(pl.LightningModule):
|
|
| 99 |
loadmodel = DefectResNet()
|
| 100 |
def load_checkpoint(checkpoint):
|
| 101 |
loadmodel.load_state_dict(checkpoint["state_dict"])
|
| 102 |
-
load_checkpoint(torch.load("
|
| 103 |
loadmodel.eval()
|
| 104 |
|
| 105 |
transform = T.Compose([
|
|
|
|
| 18 |
|
| 19 |
# Define the sample images
|
| 20 |
sample_images = {
|
| 21 |
+
"Defect01": "pics/Defect/2.jpg",
|
| 22 |
+
"Defect02": "pics/Defect/6.jpg",
|
| 23 |
+
"Defect03": "pics/Defect/8.jpg",
|
| 24 |
+
"Non-Defect01": "pics/nDefect/3.jpg",
|
| 25 |
+
"Non-Defect02": "pics/nDefect/4.jpg",
|
| 26 |
+
"Non-Defect03": "pics/nDefect/8.jpg"
|
| 27 |
}
|
| 28 |
|
| 29 |
class DefectResNet(pl.LightningModule):
|
|
|
|
| 99 |
loadmodel = DefectResNet()
|
| 100 |
def load_checkpoint(checkpoint):
|
| 101 |
loadmodel.load_state_dict(checkpoint["state_dict"])
|
| 102 |
+
load_checkpoint(torch.load("models/model.ckpt"))
|
| 103 |
loadmodel.eval()
|
| 104 |
|
| 105 |
transform = T.Compose([
|