Spaces:
Running
Running
Dharshaneshwaran
commited on
Commit
·
869a8b2
1
Parent(s):
f04c7ad
fixed the cross platform issue
Browse files- 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 =
|
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"}
|