Spaces:

rararara9999
/

facemask_detection

Sleeping

rararara9999 commited on Oct 13, 2024

Commit

68f31d1

verified ·

1 Parent(s): ad60a4b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ subprocess.check_call(["pip", "install", "evaluate"])
 subprocess.check_call(["pip", "install", "scikit-learn"])
 subprocess.check_call(["pip", "install", "torchvision"])
-model_checkpoint = "Heem2/Facemask-detection"
 batch_size = 128
 from datasets import load_dataset

 subprocess.check_call(["pip", "install", "scikit-learn"])
 subprocess.check_call(["pip", "install", "torchvision"])
+model_checkpoint = "microsoft/resnet-50"
 batch_size = 128
 from datasets import load_dataset