PickScore

Runtime error

yuvalkirstain commited on May 7, 2023

Commit

e4993da

1 Parent(s): 46535f9

try to change path to relative one

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from glob import glob
 import torch
 from transformers import AutoModel, AutoProcessor
-DEFAULT_EXAMPLE_PATH = f'{os.path.dirname(__file__)}/examples/example_0'
 device = "cuda" if torch.cuda.is_available() else "cpu"
 weight_dtype = torch.bfloat16 if device == "cuda" else torch.float32

 import torch
 from transformers import AutoModel, AutoProcessor
+DEFAULT_EXAMPLE_PATH = f'examples/example_0'
 device = "cuda" if torch.cuda.is_available() else "cpu"
 weight_dtype = torch.bfloat16 if device == "cuda" else torch.float32