sagar007 commited on
Commit
559a174
·
verified ·
1 Parent(s): d717cf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def load_model(model_path):
108
  model.eval()
109
  return model
110
 
111
- model = load_model('GPT_model.pth') # Replace with the actual path to your .pth file
112
  enc = tiktoken.get_encoding('gpt2')
113
 
114
  def generate_text(prompt, max_length=100, temperature=0.7):
 
108
  model.eval()
109
  return model
110
 
111
+ model = load_model('gpt_5000.pt') # Replace with the actual path to your .pth file
112
  enc = tiktoken.get_encoding('gpt2')
113
 
114
  def generate_text(prompt, max_length=100, temperature=0.7):