Spaces:
Runtime error
Runtime error
Commit
•
8aedac7
1
Parent(s):
6258c3d
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,9 @@ import lightning.pytorch as pl
|
|
| 5 |
from torchvision import datasets
|
| 6 |
from torch.nn import functional as F
|
| 7 |
from torch.utils.data import DataLoader, Dataset, random_split
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
def generate_art(character_dropdown, seed_slider):
|
| 10 |
|
|
|
|
| 5 |
from torchvision import datasets
|
| 6 |
from torch.nn import functional as F
|
| 7 |
from torch.utils.data import DataLoader, Dataset, random_split
|
| 8 |
+
from utils import GPTLM
|
| 9 |
+
|
| 10 |
+
newmodel = GPTLM.load_from_checkpoint('/content/drive/MyDrive/ERAV1/S21/shakespeare_gpt.pth')
|
| 11 |
|
| 12 |
def generate_art(character_dropdown, seed_slider):
|
| 13 |
|