Hugging Face's logo

torinriley
/
FTCVision-PyTorch

Object Detection
Model card Files
xet
Community
torinriley commited on
Commit
a3af870
·
verified ·
1 Parent(s): c2ce655

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -3,7 +3,7 @@ from torchvision import transforms
3
  from PIL import Image
4
  import io
5
 
6
- MODEL_PATH = "model_checkpoint.pt"
7
  NUM_CLASSES = 4
8
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
9
 
 
3
  from PIL import Image
4
  import io
5
 
6
+ MODEL_PATH = "model.pt"
7
  NUM_CLASSES = 4
8
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
9