Spaces:

gustproof
/

style-similarity

Running

gustproof commited on Apr 12

Commit

2f6b614

verified ·

1 Parent(s): 8edd9a9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ print("starting...")
 print("loaded embs")
 model = torch.load(
     "style-extractor-v0.3.0.ckpt",
-    map_location="cpu",
 )
 print("loaded extractor")
 with open("urls.txt") as f:

 print("loaded embs")
 model = torch.load(
     "style-extractor-v0.3.0.ckpt",
+    map_location="cpu", weights_only=False,
 )
 print("loaded extractor")
 with open("urls.txt") as f: