Dharshaneshwaran commited on
Commit
869a8b2
·
1 Parent(s): f04c7ad

fixed the cross platform issue

Browse files
Files changed (1) hide show
  1. inference_2.py +1 -1
inference_2.py CHANGED
@@ -38,7 +38,7 @@ import os
38
  import numpy as np
39
 
40
  # Constants
41
- MODEL_PATH = r"models\ai_detector\pytorch_model.pth"
42
  IMG_SIZE = 380
43
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
44
  LABEL_MAPPING = {0: "AI-generated", 1: "Human-created"}
 
38
  import numpy as np
39
 
40
  # Constants
41
+ MODEL_PATH = os.path.join("models", "ai_detector", "pytorch_model.pth")
42
  IMG_SIZE = 380
43
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
44
  LABEL_MAPPING = {0: "AI-generated", 1: "Human-created"}