Update app.py

app.py

@@ -10,7 +10,7 @@ from shapely.geometry import Point
 import geopandas as gpd
 from geopandas import GeoDataFrame
 
-model = torch.hub.load('facebookresearch/dinov2', 'dinov2_vits14').to("cuda")
+model = torch.hub.load('facebookresearch/dinov2', 'dinov2_vits14')
 model.eval()
 
 metadata = pd.read_csv("metadatav3.csv")
@@ -57,7 +57,7 @@ def guess_image(img):
     img = img.convert('RGB')
     print(img)
     with torch.no_grad():
-        features = model(transform(img).to("cuda").unsqueeze(0))[0].cpu()
+        features = model(transform(img).unsqueeze(0))[0].cpu()
         distances, neighbors = knn.kneighbors(features.unsqueeze(0))
 
     neighbors = neighbors[0]