gihakkk commited on
Commit
735544b
·
verified ·
1 Parent(s): 21a8746

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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:\Users\연준모\Desktop\공모전\AI공모전\AI공모전 K-PASS\ver3\1. 탈락후 재시작\Ai\사진 분석 AI\MobileVIT\mobilevit_model.onnx'
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:\Users\연준모\Desktop\공모전\AI공모전\AI공모전 K-PASS\ver3\1. 탈락후 재시작\Ai\사진 분석 AI\MobileVIT\1234567.jpg'
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