Hugging Face's logo

Spaces:
AseemD
/
gpt2
Running

App Files Community
AseemD commited on
Commit
314be9c
·
1 Parent(s): 4945ae1

added the model

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. model_19072.pt +3 -0
app.py CHANGED
@@ -17,7 +17,7 @@ UNTRAINED_MODEL.eval()
17
 
18
  # Load fine-tuned model
19
  TRAINED_MODEL = GPT(GPTConfig)
20
- checkpoint = torch.load("log/model_19072.pt", weights_only=False)
21
  TRAINED_MODEL.load_state_dict(checkpoint["model"])
22
  TRAINED_MODEL.to(device)
23
  TRAINED_MODEL.eval()
 
17
 
18
  # Load fine-tuned model
19
  TRAINED_MODEL = GPT(GPTConfig)
20
+ checkpoint = torch.load("model_19072.pt", weights_only=False)
21
  TRAINED_MODEL.load_state_dict(checkpoint["model"])
22
  TRAINED_MODEL.to(device)
23
  TRAINED_MODEL.eval()
model_19072.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a436f29aeae593de88bf0e3283ece6fe9629519ad22c6477dc0d45b6a1f18f32
3
+ size 548149826