Update app.py
Browse files
app.py
CHANGED
@@ -108,7 +108,7 @@ def load_model(model_path):
|
|
108 |
model.eval()
|
109 |
return model
|
110 |
|
111 |
-
model = load_model('
|
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):
|