loaded pyg to cpu

app.py

@@ -27,9 +27,10 @@ API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffus
 # --- LOAD DATA AND MODEL ---
 # map_location forces the model to be loaded on the CPU for huggingface compatibility
 movies_df = pd.read_csv("./sampled_movie_dataset/movies_metadata.csv")  # Load your movie data
-data = torch.load("./PyGdata.pt", map_location=device('cpu'))
-
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+
+data = torch.load("./PyGdata.pt", map_location=device('cpu'))
 model = model_def.Model(hidden_channels=32).to(device) 
 model.load_state_dict(torch.load("PyGTrainedModelState.pt", map_location=device('cpu')))
 model.eval()