Update README.md

README.md

@@ -48,14 +48,12 @@ Example inference:
 # imports
 import torch
 from PIL import Image
-
 from model import LocationDecoder  # ReGeo model class: https://github.com/TobiasRothlin/GeoLocalization/blob/main/src/DGX1/src/RegressionPretraining/Model.py
 from transformers import CLIPProcessor
 
 # load custom config (do not use AutoConfig), an example can be found in this repo
 config = { ... }
 
-# 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 preprocessor = CLIPProcessor.from_pretrained('openai/clip-vit-large-patch14-336')
 model = LocationDecoder.from_pretrained('OSTswiss/ReGeo', config=config)
@@ -70,6 +68,7 @@ image = Image.open(image_path)
 model_input = preprocessor(images=image, return_tensors="pt")
 pixel_values = model_input['pixel_values'].to(device)
 
+# run inference
 with torch.no_grad():
     output = model(pixel_values)
     normal_coordinates = output.squeeze().tolist()