Update app.py

app.py

@@ -8,9 +8,13 @@ import time
 import json
 import numpy as np
 import cv2
-from inference_sdk import InferenceHTTPClient, HTTPCallErrorError
+from inference_sdk import InferenceHTTPClient
 import matplotlib.pyplot as plt
-import base64 
+import base64
+
+# 오류 처리를 위한 예외 클래스 정의
+class APIError(Exception):
+    pass
 
 # Load model and tokenizer
 @st.cache_resource
@@ -47,7 +51,7 @@ if api_key:
             # 이미지를 base64로 인코딩
             encoded_image = base64.b64encode(image_data).decode('utf-8')
             
-            # 원본 이미지 로드 (여기로 이동)
+            # 원본 이미지 로드
             image = cv2.imread(image_path)
             image = cv2.resize(image, (800, 600))
             mask = np.zeros(image.shape, dtype=np.uint8)
@@ -55,9 +59,8 @@ if api_key:
             try:
                 # Roboflow API 호출
                 results = CLIENT.infer(encoded_image, model_id="closet/1")
-            except HTTPCallErrorError as api_error:
-                st.error(f"API Error: {api_error.description}")
-                st.error(f"API Message: {api_error.api_message}")
+            except Exception as api_error:
+                st.error(f"API Error: {str(api_error)}")
                 return Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
             
             if 'predictions' in results: