Update app.py

app.py

@@ -17,20 +17,16 @@ from torchvision import transforms, models
 
 class GeometryImageClassifier:
     def __init__(self):
-        def __init__(self):
-            # Change this line
-            self.model = models.resnet18(weights=models.ResNet18_Weights.DEFAULT)  # Updated syntax
-            self.model.fc = torch.nn.Identity()
-            self.model = self.model.to('cpu')
-            self.model.eval()
-        
+        self.model = models.resnet18(weights=models.ResNet18_Weights.DEFAULT)  # Updated syntax
+        self.model.fc = torch.nn.Identity()
+        self.model = self.model.to('cpu')
+        self.model.eval()
         self.transform = transforms.Compose([
             transforms.Resize((224, 224)),
             transforms.ToTensor(),
             transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
         ])
         
-        # Simplified reference embeddings
         self.reference_embeddings = {
             "flat.png": {
                 "embedding": None,