Spaces:

shashnk
/

shakespearean-gpt

Runtime error

shashnk commited on Nov 2, 2023

Commit

ff8b503

•

1 Parent(s): cd74f24

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -157,7 +157,7 @@ class BigramLanguageModel(nn.Module):
 def generate_text(seed_text):
     # Convert seed text to tensor
-    loaded_model = torch.load("full_bigram_language_model.pth", map_location=torch.device('cpu'))
     loaded_model.to(device)
     loaded_model.eval()

 def generate_text(seed_text):
     # Convert seed text to tensor
+    loaded_model = torch.load("model.pth", map_location=torch.device('cpu'))
     loaded_model.to(device)
     loaded_model.eval()