Update app.py
Browse files
app.py
CHANGED
@@ -91,7 +91,7 @@ import json
|
|
91 |
|
92 |
import torch
|
93 |
import torch.nn as nn
|
94 |
-
|
95 |
|
96 |
#2. Loading our JSON Data
|
97 |
#from google.colab import drive #commented
|
|
|
91 |
|
92 |
import torch
|
93 |
import torch.nn as nn
|
94 |
+
from torch.utils.data import Dataset, DataLoader
|
95 |
|
96 |
#2. Loading our JSON Data
|
97 |
#from google.colab import drive #commented
|