Fix Gradio compatibility issues

app.py

@@ -1,5 +1,5 @@
 """
-한국 번호판 OCR - KLPR_v2 (Model v5)
+한국 번호판 OCR - KLPR_v1 (Model v5)
 Hugging Face Gradio App
 """
 
@@ -8,6 +8,7 @@ import torch
 import torch.nn as nn
 from PIL import Image
 import torchvision.transforms as transforms
+import numpy as np
 
 # ============================================================================
 # 모델 정의
@@ -84,10 +85,10 @@ def decode_predictions(outputs, itos, blank_idx=0):
 # ============================================================================
 def preprocess_image(image, img_height=32, max_width=200):
     """번호판 이미지 전처리"""
-    # PIL Image로 변환 (Gradio가 numpy array를 전달)
-    if not isinstance(image, Image.Image):
-        image = Image.fromarray(image)
-
+    # numpy array를 PIL Image로 변환
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image.astype('uint8'))
+    
     image = image.convert('L')
 
     # 리사이즈 (aspect ratio 유지)
@@ -156,7 +157,7 @@ def predict_license_plate(image):
 # ============================================================================
 demo = gr.Interface(
     fn=predict_license_plate,
-    inputs=gr.Image(type="numpy", label="번호판 이미지"),
+    inputs=gr.Image(),
     outputs=gr.Textbox(label="인식 결과"),
     title="🚗 한국 번호판 OCR - KLPR v2",
     description="""
@@ -171,13 +172,7 @@ demo = gr.Interface(
     **사용 방법:**
     1. 번호판 이미지를 업로드하세요
     2. 자동으로 번호판 번호가 인식됩니다
-
-    **v2의 특징:**
-    - 더 많은 지역명 지원 (산, 제, 세, 종, 파)
-    - 18 epoch 훈련으로 더 안정적인 학습
     """,
-    examples=None,
-    cache_examples=False,
 )
 
 if __name__ == "__main__":

requirements.txt

@@ -1,4 +1,4 @@
+gradio==4.44.0
 torch
 torchvision
-gradio
 Pillow