Update README.md
Browse files
README.md
CHANGED
@@ -35,7 +35,7 @@ import torch
|
|
35 |
import torchvision.transforms as transforms
|
36 |
from transformers import AutoModel
|
37 |
|
38 |
-
model_tag = f"
|
39 |
model = AutoModel.from_pretrained(model_tag, trust_remote_code=True)
|
40 |
|
41 |
def generate_random_image(height=440, width=440, channels=3):
|
|
|
35 |
import torchvision.transforms as transforms
|
36 |
from transformers import AutoModel
|
37 |
|
38 |
+
model_tag = f"conservationxlabs/miewid-msv2"
|
39 |
model = AutoModel.from_pretrained(model_tag, trust_remote_code=True)
|
40 |
|
41 |
def generate_random_image(height=440, width=440, channels=3):
|