Update app.py

app.py

@@ -9,15 +9,15 @@ def load_model():
     """Load the trained GPT model"""
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     GPT_CONFIG = {
-        vocab_size = 50257
-        n_heads = 8
-        n_layers = 6
-        head_size = 64
-        n_embd = 512
-        block_size = 128
-        dropout = 0.1
-        learning_rate = 3e-4
-        weight_decay = 0.1
+        "vocab_size" : 50257,
+        "n_heads" : 8,
+        "n_layers" : 6,
+        "head_size" : 64,
+        "n_embd" : 512,
+        "block_size" : 128,
+        "dropout" : 0.1,
+        "learning_rate" : 3e-4,
+        "weight_decay" : 0.1,
     }
     model = GPTLanguageModel(GPT_CONFIG)
     model.load_state_dict(torch.load("model_weights.pth", map_location=device))