Update to work on mac

.python-version

@@ -0,0 +1 @@
+3.10

main.py

@@ -64,7 +64,7 @@ def run_model(model: nn.Module, response_size: int = BLOCK_SIZE):
 
 if os.path.exists('model.pth'):
     print("Loading model from file...")
-    checkpoint = torch.load('model.pth')
+    checkpoint = torch.load('model.pth', map_location=DEVICE)
     m.load_state_dict(checkpoint['model_state_dict'])
     print("Model loaded!")
 else: