Spaces:

kishkath
/

GPT2

Sleeping

kishkath commited on Jan 24

Commit

1152b2a

verified ·

1 Parent(s): f47b4c5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ def initialize_model():
     global model
     if model is None:
         model = GPT(GPTConfig())
-        model.load_state_dict(torch.load('model_state_dict.pth', map_location=torch.device('cpu')))
         model.eval()
     return model

     global model
     if model is None:
         model = GPT(GPTConfig())
+        model.load_state_dict(torch.load('models/model_state_dict.pth', map_location=torch.device('cpu')))
         model.eval()
     return model