Update app.py

app.py

@@ -31,8 +31,7 @@ def preprocess_image(image):
     
     return image
 
-# caption_image 함수 내에서 사용
-pil_image = preprocess_image(pil_image)
+# REMOVE THIS LINE: pil_image = preprocess_image(pil_image)
 
 def patched_json_schema(schema, defs=None):
     # Handle boolean schema directly
@@ -197,9 +196,8 @@ def caption_image(image, model_name=default_caption_model):
             print(f"Unexpected image type: {type(image)}")
             return "Error: Unsupported image type"
     
-    # Convert input to RGB if needed
-    if pil_image.mode != "RGB":
-        pil_image = pil_image.convert("RGB")
+    # Apply preprocessing to the PIL image (use the preprocess_image function here)
+    pil_image = preprocess_image(pil_image)
     
     # Check if model is available
     if model_name not in models or model_name not in processors:
@@ -316,7 +314,6 @@ custom_css = """
 .gradio-container {
     max-width: 1200px !important;
 }
-
 .main-header {
     text-align: center;
     margin-bottom: 1rem;
@@ -327,27 +324,23 @@ custom_css = """
     -webkit-text-fill-color: transparent;
     padding: 0.5rem;
 }
-
 .tagline {
     text-align: center;
     font-size: 1.2rem;
     margin-bottom: 2rem;
     color: #4b5563;
 }
-
 .image-preview {
     border-radius: 12px;
     overflow: hidden;
     box-shadow: 0 10px 15px -3px rgb(0 0 0 / 0.1), 0 4px 6px -4px rgb(0 0 0 / 0.1);
 }
-
 .panel-box {
     border-radius: 12px;
     background-color: rgba(255, 255, 255, 0.8);
     padding: 1rem;
     box-shadow: 0 4px 6px -1px rgb(0 0 0 / 0.1), 0 2px 4px -2px rgb(0 0 0 / 0.1);
 }
-
 .control-panel {
     padding: 1rem;
     border-radius: 12px;
@@ -355,26 +348,22 @@ custom_css = """
     margin-bottom: 1rem;
     border: 1px solid #e2e8f0;
 }
-
 .section-header {
     font-weight: 600;
     font-size: 1.1rem;
     margin-bottom: 0.5rem;
     color: #4338ca;
 }
-
 .transform-button {
     font-weight: 600 !important;
     margin-top: 1rem !important;
 }
-
 .footer {
     text-align: center;
     color: #6b7280;
     margin-top: 2rem;
     font-size: 0.9rem;
 }
-
 .output-panel {
     background: linear-gradient(135deg, #f0f9ff, #e0f2fe);
     border-radius: 12px;