Update app.py

app.py

@@ -92,7 +92,7 @@ def load_model(model_name):
     print(f"Loading {model_size} model: {model_name}")
     
     model_path = os.path.join(MODELS_DIR, model_name)
-    global_model = build_model(model_size, device="cpu").to(device)
+    global_model = build_model(model_size).to(device)
     
     try:
         state_dict = torch.load(model_path, map_location=device, weights_only=True)