lashao commited on
Commit
85cc80e
1 Parent(s): dda4e58

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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"lashao/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):
 
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):