Update README.md
Browse files
README.md
CHANGED
@@ -50,7 +50,7 @@ import torch
|
|
50 |
from transformers import RobertaTokenizer
|
51 |
from torchvision import transforms
|
52 |
from PIL import Image
|
53 |
-
from
|
54 |
|
55 |
# Load model
|
56 |
model = LVL()
|
|
|
50 |
from transformers import RobertaTokenizer
|
51 |
from torchvision import transforms
|
52 |
from PIL import Image
|
53 |
+
from model import LVL # Replace with your module or package
|
54 |
|
55 |
# Load model
|
56 |
model = LVL()
|