Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -157,7 +157,7 @@ class BigramLanguageModel(nn.Module):
|
|
157 |
|
158 |
def generate_text(seed_text):
|
159 |
# Convert seed text to tensor
|
160 |
-
loaded_model = torch.load("
|
161 |
loaded_model.to(device)
|
162 |
loaded_model.eval()
|
163 |
|
|
|
157 |
|
158 |
def generate_text(seed_text):
|
159 |
# Convert seed text to tensor
|
160 |
+
loaded_model = torch.load("model.pth", map_location=torch.device('cpu'))
|
161 |
loaded_model.to(device)
|
162 |
loaded_model.eval()
|
163 |
|