Hugging Face's logo

Spaces:
Muinez
/
Image-scorer
Sleeping

App Files Community
Muinez commited on
Commit
3206424
·
verified ·
1 Parent(s): 1d56378

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ model_file = hf_hub_download(repo_id="Muinez/Image-scorer", filename="scorer.pth
32
 
33
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
34
  model = ScoreClassifier().to(DEVICE)
35
- model.load_state_dict(torch.load("scorer.pth"))
36
  model.eval()
37
 
38
  processor = SiglipImageProcessor.from_pretrained('google/siglip-base-patch16-512')
 
32
 
33
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
34
  model = ScoreClassifier().to(DEVICE)
35
+ model.load_state_dict(torch.load(model_file))
36
  model.eval()
37
 
38
  processor = SiglipImageProcessor.from_pretrained('google/siglip-base-patch16-512')