Update README.md
Browse files
README.md
CHANGED
@@ -10,7 +10,7 @@ from transformers import AutoFeatureExtractor
|
|
10 |
from PIL import Image
|
11 |
|
12 |
# ONNX 모델 경로
|
13 |
-
onnx_model_path = r'C:\
|
14 |
|
15 |
# ONNX 런타임 세션 초기화
|
16 |
ort_session = ort.InferenceSession(onnx_model_path)
|
@@ -38,7 +38,7 @@ def predict_image(image_path):
|
|
38 |
return "그냥 사진" if predicted_class == 1 else "로맨스 스캠 사진"
|
39 |
|
40 |
# 예측 예시
|
41 |
-
image_path = r'C:\
|
42 |
result = predict_image(image_path)
|
43 |
print(result)
|
44 |
|
|
|
10 |
from PIL import Image
|
11 |
|
12 |
# ONNX 모델 경로
|
13 |
+
onnx_model_path = r'C:\mobilevit_model.onnx'
|
14 |
|
15 |
# ONNX 런타임 세션 초기화
|
16 |
ort_session = ort.InferenceSession(onnx_model_path)
|
|
|
38 |
return "그냥 사진" if predicted_class == 1 else "로맨스 스캠 사진"
|
39 |
|
40 |
# 예측 예시
|
41 |
+
image_path = r'C:\1234567.jpg'
|
42 |
result = predict_image(image_path)
|
43 |
print(result)
|
44 |
|