Update backend.py

backend.py

@@ -1,13 +1,3 @@
-from huggingface_hub import InferenceClient
-IS_SPACES_ZERO = os.environ.get("SPACES_ZERO_GPU", "0") == "1"
-IS_SPACE = os.environ.get("SPACE_ID", None) is not None
-
-device = "cuda" if torch.cuda.is_available() else "cpu"
-LOW_MEMORY = os.getenv("LOW_MEMORY", "0") == "1"
-print(f"Using device: {device}")
-print(f"low memory: {LOW_MEMORY}")
-
-
 import os
 import pandas as pd
 import requests
@@ -21,6 +11,9 @@ import textwrap
 from haversine import haversine
 from geopy.geocoders import Nominatim
 from huggingface_hub import InferenceClient
+import os
+IS_SPACES_ZERO = os.environ.get("SPACES_ZERO_GPU", "0") == "1"
+IS_SPACE = os.environ.get("SPACE_ID", None) is not None
 
 # Constants
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"