Hugging Face's logo

Spaces:
yuu1234
/
SFW
Sleeping

App Files Community
yuu1234 commited on
Commit
142180a
·
1 Parent(s): bd48ebf

Change model

Browse files
Files changed (3) hide show
  1. app.py +1 -1
  2. resnet50_best_9818.pth +3 -0
  3. resnet50_best_9838.pth +3 -0
app.py CHANGED
@@ -17,7 +17,7 @@ import gradio as gr
17
  DEVICE = torch.device("cpu") # HF free → CPU
18
  NUM_CLASSES = 3
19
  IMG_SIZE = 224
20
- MODEL_PATH = "resnet50_best_2.pth"
21
 
22
  class_mapping = {
23
  0: "safe",
 
17
  DEVICE = torch.device("cpu") # HF free → CPU
18
  NUM_CLASSES = 3
19
  IMG_SIZE = 224
20
+ MODEL_PATH = "resnet50_best_9838.pth"
21
 
22
  class_mapping = {
23
  0: "safe",
resnet50_best_9818.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a894754a933b04553b9030edd550693c767a84eba4c9e958ce7a6e7219439947
3
+ size 94376739
resnet50_best_9838.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d20e2cc82293e97f9f9908785e249076684bb9361f9b6ff31d1e0c9dab6dd7e
3
+ size 94376739