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"}
|