Update app.py

app.py

@@ -11,13 +11,18 @@ from langchain.memory import ConversationBufferMemory
 import torch
 from PIL import Image
 from torchvision import transforms, models
+from langchain_community.llms import HuggingFaceEndpoint
+from langchain_huggingface import HuggingFaceEmbeddings
+from torchvision import transforms, models
 
 class GeometryImageClassifier:
     def __init__(self):
-        self.model = models.resnet18(pretrained=True)
-        self.model.fc = torch.nn.Identity()
-        self.model = self.model.to('cpu')
-        self.model.eval()
+        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.transform = transforms.Compose([
             transforms.Resize((224, 224)),