Upload app.py

app.py

@@ -44,7 +44,7 @@ def load_model():
     
     #model.load_state_dict(torch.load(checkpoint_path, map_location=torch.device('cpu')))
     checkpoint = torch.load(checkpoint_path, map_location=device)
-    config = checkpoint['config']
+    #config = checkpoint['config']
     model = SmolLM(vocab_size, embed_dim, num_heads, num_layers, max_seq_len)
     model.load_state_dict(checkpoint['model_state_dict'])
     model.to(device)