rararara9999 commited on
Commit
b2ef3fb
·
verified ·
1 Parent(s): b1410b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ subprocess.check_call(["pip", "install", "scikit-learn"])
10
  subprocess.check_call(["pip", "install", "torchvision"])
11
 
12
 
13
- model_checkpoint = "Heem2/Facemask-detection"
14
  batch_size = 128
15
 
16
  from datasets import load_dataset
 
10
  subprocess.check_call(["pip", "install", "torchvision"])
11
 
12
 
13
+ model_checkpoint = "microsoft/resnet-50"
14
  batch_size = 128
15
 
16
  from datasets import load_dataset