hotfix device

app.py

@@ -4,13 +4,13 @@ from transformers import AutoTokenizer
 
 from model import GPT, GPTConfig
 
-DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 
 def setup(model_path: str):
     tokenizer = AutoTokenizer.from_pretrained("gpt2")
     if DEVICE == "cpu":
-        checkpoint = torch.load(model_path, map_location=torch.device('cpu'))
+        checkpoint = torch.load(model_path, map_location=torch.device("cpu"))
     else:
         checkpoint = torch.load(model_path)
     model = GPT(GPTConfig(**checkpoint["model_args"]))