Hugging Face's logo

Spaces:
anilbhatt1
/
GPT-Passage-Generator
Sleeping

App Files Community
anilbhatt1 commited on
Commit
43dd287
•
1 Parent(s): 5d58041

Modified req

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. checkpoint_epoch-199999_26.10.2023_13_20_27_cpu.pt → checkpoint_cpu.pt +0 -0
app.py CHANGED
@@ -150,7 +150,7 @@ class gptModel(nn.Module):
150
 
151
  model = gptModel()
152
  m = model.to(device)
153
- model_pth = 'checkpoint_epoch-199999_26.10.2023_13:20:27_cpu.pt'
154
  model.load_state_dict(torch.load(model_pth))
155
 
156
  def generate_text(given_text_context, max_text_length):
 
150
 
151
  model = gptModel()
152
  m = model.to(device)
153
+ model_pth = './checkpoint_cpu.pt'
154
  model.load_state_dict(torch.load(model_pth))
155
 
156
  def generate_text(given_text_context, max_text_length):
checkpoint_epoch-199999_26.10.2023_13_20_27_cpu.pt → checkpoint_cpu.pt RENAMED
File without changes