Spaces:

fashion-demo-organization
/

fashion_demo

Running

simonhermansson commited on Apr 20, 2023

Commit

52a1718

1 Parent(s): 8ad8e13

Add correct path to checkpoint.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import pandas as pd
 # Load model
-checkpoint_path = "../finetuned.pth"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model, preprocess = clip.load(checkpoint_path, device=device, jit=False)

 # Load model
+checkpoint_path = "files/finetuned.pth"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model, preprocess = clip.load(checkpoint_path, device=device, jit=False)