Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -31,7 +31,7 @@ def predict(model, image_tensor, device, threshold=0.5):
|
|
| 31 |
|
| 32 |
if __name__ == "__main__":
|
| 33 |
model_path = r"model.pt" # Path to Flux-Detector
|
| 34 |
-
image_path = r"test_image.
|
| 35 |
|
| 36 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 37 |
model = load_model(model_path, device)
|
|
|
|
| 31 |
|
| 32 |
if __name__ == "__main__":
|
| 33 |
model_path = r"model.pt" # Path to Flux-Detector
|
| 34 |
+
image_path = r"test_image.png" # Path to test image
|
| 35 |
|
| 36 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 37 |
model = load_model(model_path, device)
|