Update app.py

app.py

@@ -54,8 +54,17 @@ device = torch.device("cpu")
 
 testing_df = pd.read_csv("testing_df.csv")
 model = CLIPModel()  # Create an instance of CLIPModel
-model.load_state_dict(torch.load("weights.pt", map_location=torch.device('cpu')))
-# ...)
+# Load the model
+state_dict = torch.load("weights.pt", map_location=torch.device('cpu'))
+print("Loaded State Dict Keys:", state_dict.keys())
+
+# Create an instance of CLIPModel
+model = CLIPModel().to(device)
+print("Model Keys:", model.state_dict().keys())
+
+# Load the state_dict into the model
+model.load_state_dict(state_dict, strict=False)  # Set strict=False to ignore unexpected keys
+
 text_embeddings = torch.load('saved_text_embeddings.pt', map_location=device)
 
 def download_link(content, filename, link_text):