Spaces:

Jack000
/

glid-3-xl-stable-classifier

Runtime error

Jack000 commited on Sep 17, 2022

Commit

66b8a61

1 Parent(s): 0d15c14

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -120,7 +120,7 @@ classifier_photo.eval()
 classifier_art = create_classifier(**classifier_config)
 classifier_art.load_state_dict(
-    torch.load('https://huggingface.co/Jack000/glid-3-xl-stable/resolve/main/classifier_art/model110000.pt', map_location="cpu")
 )
 classifier_art.to(device)
 classifier_art.convert_to_fp16()

 classifier_art = create_classifier(**classifier_config)
 classifier_art.load_state_dict(
+    torch.load(fetch('https://huggingface.co/Jack000/glid-3-xl-stable/resolve/main/classifier_art/model110000.pt'), map_location="cpu")
 )
 classifier_art.to(device)
 classifier_art.convert_to_fp16()