eva-vit-1b-224-8043-gradio

Running

cdnuts commited on Feb 29

Commit

0a34073

•

1 Parent(s): b90255f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 from torchvision.transforms import transforms
 import gradio as gr
 from huggingface_hub import snapshot_download
-snapshot_download(repo_id="Thouph/eva-vit-1b-224-8043", local_dir = "")
 model = torch.load('model.pth', map_location=torch.device('cpu'))

 from torchvision.transforms import transforms
 import gradio as gr
 from huggingface_hub import snapshot_download
+snapshot_download(repo_id="Thouph/eva-vit-1b-224-8043", local_dir = "./")
 model = torch.load('model.pth', map_location=torch.device('cpu'))