torinriley
/

FTCVision-PyTorch

Object Detection

Inference Endpoints

Model card Files Files and versions Community

torinriley commited on 12 days ago

Commit

a3af870

•

1 Parent(s): c2ce655

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -3,7 +3,7 @@ from torchvision import transforms
 from PIL import Image
 import io
-MODEL_PATH = "model_checkpoint.pt"
 NUM_CLASSES = 4
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

 from PIL import Image
 import io
+MODEL_PATH = "model.pt"
 NUM_CLASSES = 4
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"