Update nygaardcodecommentclassification/api/controllers.py

Removed helper method

Using torch.set_default_device

Added map_location for loading on cpu

Update nygaardcodecommentclassification/api/models.py

Update nygaardcodecommentclassification/api/models.py

First commit