Hugging Face's logo

Spaces:
anilbhatt1
/
GPT-Passage-Generator
Sleeping

App Files Community
anilbhatt1 commited on
Commit
6c4b61b
•
1 Parent(s): e27163e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -152,6 +152,7 @@ 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):
157
 
 
152
  m = model.to(device)
153
  model_pth = './checkpoint_cpu.pt'
154
  model.load_state_dict(torch.load(model_pth))
155
+ print('Model loaded')
156
 
157
  def generate_text(given_text_context, max_text_length):
158