Update app.py

app.py

@@ -1,3 +1,103 @@
+import os
+import sys
+from torchvision.transforms import functional
+sys.modules["torchvision.transforms.functional_tensor"] = functional
+
+from basicsr.archs.srvgg_arch import SRVGGNetCompact
+from gfpgan.utils import GFPGANer
+from realesrgan.utils import RealESRGANer
+
+import torch
+import cv2
+import gradio as gr
+
+# 필요한 모델 다운로드
+def download_upscaler_models():
+    if not os.path.exists('realesr-general-x4v3.pth'):
+        os.system("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth -P .")
+    if not os.path.exists('GFPGANv1.4.pth'):
+        os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth -P .")
+
+# 업스케일러 모델 초기화
+def init_upscaler_model():
+    model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
+    model_path = 'realesr-general-x4v3.pth'
+    half = True if torch.cuda.is_available() else False
+    upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
+    
+    return upsampler
+
+# 이미지 업스케일링 함수
+def upscale_image(img):
+    try:
+        upsampler = init_upscaler_model()
+        
+        # 이미지 파일 읽기
+        img = cv2.imread(img, cv2.IMREAD_UNCHANGED)
+        if len(img.shape) == 3 and img.shape[2] == 4:
+            img_mode = 'RGBA'
+        elif len(img.shape) == 2:
+            img_mode = None
+            img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
+        else:
+            img_mode = None
+
+        # 작은 이미지 크기 조정
+        h, w = img.shape[0:2]
+        if h < 300:
+            img = cv2.resize(img, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
+
+        # 고정된 GFPGANv1.4 모델 사용
+        face_enhancer = GFPGANer(
+            model_path='GFPGANv1.4.pth', 
+            upscale=2, 
+            arch='clean',
+            channel_multiplier=2,
+            bg_upsampler=upsampler
+        )
+
+        # 이미지 향상
+        try:
+            _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True)
+        except RuntimeError as error:
+            print('오류', error)
+            return None
+
+        # RGB로 변환하여 반환
+        output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
+        return output
+    except Exception as error:
+        print('글로벌 예외', error)
+        return None
+
+# 업스케일러 탭 UI 생성 함수
+def create_upscaler_tab():
+    # 필요한 모델 다운로드
+    download_upscaler_models()
+    
+    with gr.Tab("이미지 업스케일러"):
+        gr.Markdown(
+            """
+            # 이미지 업스케일러
+            
+            업로드한 이미지를 고해상도로 변환합니다. 특히 인물 사진의 얼굴 디테일을 향상시킵니다.
+            """
+        )
+        
+        with gr.Row():
+            with gr.Column():
+                input_image = gr.Image(type="filepath", label="입력 이미지")
+                upscale_btn = gr.Button("이미지 업스케일", variant="primary")
+            
+            with gr.Column():
+                output_image = gr.Image(type="numpy", label="결과 이미지")
+        
+        upscale_btn.click(
+            fn=upscale_image,
+            inputs=[input_image],
+            outputs=[output_image]
+        )
+
 import os
 import tempfile
 from PIL import Image
@@ -7,6 +107,18 @@ import re
 from io import BytesIO
 import time
 
+# 업스케일러 모듈 임포트
+import sys
+from torchvision.transforms import functional
+sys.modules["torchvision.transforms.functional_tensor"] = functional
+
+from basicsr.archs.srvgg_arch import SRVGGNetCompact
+from gfpgan.utils import GFPGANer
+from realesrgan.utils import RealESRGANer
+
+import torch
+import cv2
+
 from google import genai
 from google.genai import types
 
@@ -299,140 +411,223 @@ def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progres
     
     return results[0], results[1], results[2], results[3], combined_status, combined_prompts
 
+# 이미지 업스케일러 기능
+def download_upscaler_models():
+    if not os.path.exists('realesr-general-x4v3.pth'):
+        os.system("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth -P .")
+    if not os.path.exists('GFPGANv1.4.pth'):
+        os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth -P .")
+
+def upscale_image(img):
+    try:
+        # 모델 초기화
+        model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
+        model_path = 'realesr-general-x4v3.pth'
+        half = True if torch.cuda.is_available() else False
+        upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
+        
+        # 이미지 파일 읽기
+        img = cv2.imread(img, cv2.IMREAD_UNCHANGED)
+        if len(img.shape) == 3 and img.shape[2] == 4:
+            img_mode = 'RGBA'
+        elif len(img.shape) == 2:
+            img_mode = None
+            img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
+        else:
+            img_mode = None
+
+        # 작은 이미지 크기 조정
+        h, w = img.shape[0:2]
+        if h < 300:
+            img = cv2.resize(img, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
+
+        # GFPGANv1.4 모델 사용
+        face_enhancer = GFPGANer(
+            model_path='GFPGANv1.4.pth', 
+            upscale=2, 
+            arch='clean',
+            channel_multiplier=2,
+            bg_upsampler=upsampler
+        )
+
+        # 이미지 향상
+        try:
+            _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True)
+        except RuntimeError as error:
+            print('오류', error)
+            return None
+
+        # RGB로 변환하여 반환
+        output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
+        return output
+    except Exception as error:
+        print('글로벌 예외', error)
+        return None
+
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
-            <h1>이커머스용 이미지 생성기</h1>
-            <p>이미지를 업로드하고 예제를 참가하여 프롬프트를 수정하고 "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.</p>
+            <h1>이커머스용 이미지 생성 및 편집 도구</h1>
+            <p>이미지를 업로드하고 예제를 참고하여 프롬프트를 수정하거나 이미지 업스케일 기능을 이용하세요.</p>
         </div>
         """
     )
     
-    with gr.Row():
-        # 왼쪽 입력 영역 (예제 이미지를 포함하여 사용 방법 아래에 배치)
-        with gr.Column(scale=1):
+    with gr.Tabs():
+        # 첫 번째 탭: 이미지 생성기
+        with gr.Tab("이미지 생성기"):
             with gr.Row():
-                image1_input = gr.Image(type="pil", label="#1", image_mode="RGB", height=300, width=200)
-                image2_input = gr.Image(type="pil", label="#2", image_mode="RGB", height=300, width=200)
-                image3_input = gr.Image(type="pil", label="#3", image_mode="RGB", height=300, width=200)
-            prompt_input = gr.Textbox(
-                lines=3,
-                placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
-                label="프롬프트 (선택 사항)"
+                # 왼쪽 입력 영역 (예제 이미지를 포함하여 사용 방법 아래에 배치)
+                with gr.Column(scale=1):
+                    with gr.Row():
+                        image1_input = gr.Image(type="pil", label="#1", image_mode="RGB", height=300, width=200)
+                        image2_input = gr.Image(type="pil", label="#2", image_mode="RGB", height=300, width=200)
+                        image3_input = gr.Image(type="pil", label="#3", image_mode="RGB", height=300, width=200)
+                    prompt_input = gr.Textbox(
+                        lines=3,
+                        placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
+                        label="프롬프트 (선택 사항)"
+                    )
+                    with gr.Row():
+                        image_change_btn1 = gr.Button("이미지 변경-1")
+                        image_change_btn2 = gr.Button("이미지 변경-2")
+                        text_remove_btn = gr.Button("글자 지우기")
+                        text_change_btn = gr.Button("글자 변경하기")
+                        clothes_change_btn1 = gr.Button("가상 상품착용-1")
+                        clothes_change_btn2 = gr.Button("가상 상품착용-2")
+                        holding_product_btn = gr.Button("상품들고 있기")
+                        background_change_btn = gr.Button("배경 바꾸기")
+                        composite_product_btn = gr.Button("부분 지우기")
+                    submit_btn = gr.Button("이미지 생성 (4장)", variant="primary")
+                    
+                    gr.Markdown(
+                        """
+                        ### 사용 방법:
+                        
+                        1. **자동 합성**: 이미지를 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
+                        2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
+                        3. **선택 옵션**: 위의 버튼을 클릭하면 프롬프트 입력란에 한국어 문구로 입력하시면 됩니다.
+                        4. **다양한 이미지**: "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.
+                        5. **예제선택**: 다양한 예제를 통해 미리 테스트해보세요.
+                        
+                        > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
+                        """
+                    )
+                    
+                    # 예제 이미지: 사용 방법 아래 왼쪽 영역에 배치
+                    gr.Markdown("## 예제 이미지")
+                    examples = [
+                        ["down/모델.jpg", None, None, "(#1의 여성)이 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라."],
+                        ["down/상어레고모형.png", None, None, "(#1 레모모형)에서 청색상어레고만 검은색 고래레고로 변경하고 나머지 부분은 seed를 변경을 하지마라."],
+                        ["down/중국어.png", None, None, "(#1 이미지)에 있는 중국어를 모두 제거하라."],
+                        ["down/텍스트.webp", None, None, '(#1의 텍스트)를 스타일을 유지한체 텍스트만 "Hello"로 바꿔라'],
+                        ["down/모델.jpg", "down/선글라스.png", "down/청바지.png", "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라."],
+                        ["down/모델.jpg", "down/선글라스.png", "down/카페전경.png", "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)를 직접 모델이 착용한 것처럼 (#3의 장소)에서 의자에 앉아 있는 자연스러운 이미지를 생성하라."],
+                        ["down/모델.jpg", "down/와인잔.png", None, "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 와인잔)을 여성모델이 홍보할 와인잔을 돋보이게 들고 있는 자연스러운 모습으로 이미지를 생성하라."],
+                        ["down/모델.jpg", "down/카페전경.png", None, "(#1의 여성모델)이 (#2 이미지의 배경)을 주요 피사체는 그대로 유지하여 이미지의 분위기가 자연스럽게 어우러지도록 생성하라."],
+                        ["down/상어레고모형.png", None, None, "(#1의 레고모형)에서 청색상어레고를 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우러지도록 채워주세요. 단, 이미지의 다른 부분의 주요 요소는 동일하게 유지해 해야한다."]
+                    ]
+                    
+                    gr.Examples(
+                        examples=examples,
+                        inputs=[image1_input, image2_input, image3_input, prompt_input],
+                        elem_id="examples-grid"
+                    )
+                    
+                # 오른쪽 출력 영역
+                with gr.Column(scale=1):
+                    with gr.Row():
+                        with gr.Column():
+                            output_image1 = gr.Image(label="생성된 이미지 #1", height=600, width=450)
+                            output_image3 = gr.Image(label="생성된 이미지 #3", height=600, width=450)
+                        with gr.Column():
+                            output_image2 = gr.Image(label="생성된 이미지 #2", height=600, width=450)
+                            output_image4 = gr.Image(label="생성된 이미지 #4", height=600, width=450)
+                    
+                    output_text = gr.Textbox(label="상태 메시지", lines=4)
+                    prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True, lines=4)
+            
+            # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
+            image_change_btn1.click(
+                fn=lambda: "(#1의 여성)이 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라.",
+                inputs=[],
+                outputs=prompt_input
+            )
+            image_change_btn2.click(
+                fn=lambda: "(#1 레모모형)에서 청색상어레고만 검은색 고래레고로 변경하고 나머지 부분은 seed를 변경을 하지마라.",
+                inputs=[],
+                outputs=prompt_input
+            )
+            text_remove_btn.click(
+                fn=lambda: "(#1 이미지)에 있는 중국어를 모두 제거하라.",
+                inputs=[],
+                outputs=prompt_input
+            )
+            text_change_btn.click(
+                fn=lambda: "(#1의 텍스트)를 스타일을 유지한체 텍스트만 \"Hello\"로 바꿔라",
+                inputs=[],
+                outputs=prompt_input
+            )
+            clothes_change_btn1.click(
+                fn=lambda: "(#1의 여성모겔)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라.",
+                inputs=[],
+                outputs=prompt_input
+            )
+            clothes_change_btn2.click(
+                fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)를 직접 모델이 착용한 것처럼 (#3의 장소)에서 의자에 앉아 있는 자연스러운 이미지를 생성하라.",
+                inputs=[],
+                outputs=prompt_input
+            )
+            holding_product_btn.click(
+                fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 와인잔)을 여성모델이 홍보할 와인잔을 돋보이게 들고 있는 자연스러운 모습으로 이미지를 생성하라.",
+                inputs=[],
+                outputs=prompt_input
+            )
+            background_change_btn.click(
+                fn=lambda: "(#1의 여성모델)이 (#2 이미지의 배경)을 주요 피사체는 그대로 유지하여 두 이미지의 분위기가 자연스럽게 어우러지도록 생성하라.",
+                inputs=[],
+                outputs=prompt_input
+            )
+            composite_product_btn.click(
+                fn=lambda: "(#1의 레고모형)에서 청색상어레고를 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우러지도록 채워주세요. 단, 이미지의 다른 부분의 주요 요소는 동일하게 유지해 해야한다.",
+                inputs=[],
+                outputs=prompt_input
             )
-            with gr.Row():
-                image_change_btn1 = gr.Button("이미지 변경-1")
-                image_change_btn2 = gr.Button("이미지 변경-2")
-                text_remove_btn = gr.Button("글자 지우기")
-                text_change_btn = gr.Button("글자 변경하기")
-                clothes_change_btn1 = gr.Button("가상 상품착용-1")
-                clothes_change_btn2 = gr.Button("가상 상품착용-2")
-                holding_product_btn = gr.Button("상품들고 있기")
-                background_change_btn = gr.Button("배경 바꾸기")
-                composite_product_btn = gr.Button("부분 지우기")
-            submit_btn = gr.Button("이미지 생성 (4장)", variant="primary")
+            
+            submit_btn.click(
+                fn=generate_multiple_images,
+                inputs=[image1_input, image2_input, image3_input, prompt_input],
+                outputs=[output_image1, output_image2, output_image3, output_image4, output_text, prompt_display],
+            )
+        
+        # 두 번째 탭: 이미지 업스케일러
+        with gr.Tab("이미지 업스케일러"):
+            # 필요한 모델 다운로드 호출
+            download_upscaler_models()
             
             gr.Markdown(
                 """
-                ### 사용 방법:
-                
-                1. **자동 합성**: 이미지를 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
-                2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
-                3. **선택 옵션**: 위의 버튼을 클릭하면 프롬프트 입력란에 한국어 문구로 입력하시면 됩니다.
-                4. **다양한 이미지**: "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.
-                5. **예제선택**: 다양한 예제를 통해 미리 테스트해보세요.
+                # 이미지 업스케일러
                 
-                > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
+                업로드한 이미지를 고해상도로 변환합니다. 특히 인물 사진의 얼굴 디테일을 향상시킵니다.
                 """
             )
             
-            # 예제 이미지: 사용 방법 아래 왼쪽 영역에 배치
-            gr.Markdown("## 예제 이미지")
-            examples = [
-                ["down/모델.jpg", None, None, "(#1의 여성)이 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라."],
-                ["down/상어레고모형.png", None, None, "(#1 레모모형)에서 청색상어레고만 검은색 고래레고로 변경하고 나머지 부분은 seed를 변경을 하지마라."],
-                ["down/중국어.png", None, None, "(#1 이미지)에 있는 중국어를 모두 제거하라."],
-                ["down/텍스트.webp", None, None, '(#1의 텍스트)를 스타일을 유지한체 텍스트만 "Hello"로 바꿔라'],
-                ["down/모델.jpg", "down/선글라스.png", "down/청바지.png", "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라."],
-                ["down/모델.jpg", "down/선글라스.png", "down/카페전경.png", "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)를 직접 모델이 착용한 것처럼 (#3의 장소)에서 의자에 앉아 있는 자연스러운 이미지를 생성하라."],
-                ["down/모델.jpg", "down/와인잔.png", None, "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 와인잔)을 여성모델이 홍보할 와인잔을 돋보이게 들고 있는 자연스러운 모습으로 이미지를 생성하라."],
-                ["down/모델.jpg", "down/카페전경.png", None, "(#1의 여성모델)이 (#2 이미지의 배경)을 주요 피사체는 그대로 유지하여 이미지의 분위기가 자연스럽게 어우러지도록 생성하라."],
-                ["down/상어레고모형.png", None, None, "(#1의 레고모형)에서 청색상어레고를 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우러지도록 채워주세요. 단, 이미지의 다른 부분의 주요 요소는 동일하게 유지해 해야한다."]
-            ]
-            
-            gr.Examples(
-                examples=examples,
-                inputs=[image1_input, image2_input, image3_input, prompt_input],
-                elem_id="examples-grid"
-            )
-            
-        # 오른쪽 출력 영역
-        with gr.Column(scale=1):
             with gr.Row():
                 with gr.Column():
-                    output_image1 = gr.Image(label="생성된 이미지 #1", height=600, width=450)
-                    output_image3 = gr.Image(label="생성된 이미지 #3", height=600, width=450)
+                    input_upscale_image = gr.Image(type="filepath", label="입력 이미지")
+                    upscale_btn = gr.Button("이미지 업스케일", variant="primary")
+                
                 with gr.Column():
-                    output_image2 = gr.Image(label="생성된 이미지 #2", height=600, width=450)
-                    output_image4 = gr.Image(label="생성된 이미지 #4", height=600, width=450)
+                    output_upscale_image = gr.Image(type="numpy", label="결과 이미지")
             
-            output_text = gr.Textbox(label="상태 메시지", lines=4)
-            prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True, lines=4)
-    
-    # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
-    image_change_btn1.click(
-        fn=lambda: "(#1의 여성)이 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    image_change_btn2.click(
-        fn=lambda: "(#1 레모모형)에서 청색상어레고만 검은색 고래레고로 변경하고 나머지 부분은 seed를 변경을 하지마라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    text_remove_btn.click(
-        fn=lambda: "(#1 이미지)에 있는 중국어를 모두 제거하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    text_change_btn.click(
-        fn=lambda: "(#1의 텍스트)를 스타일을 유지한체 텍스트만 \"Hello\"로 바꿔라",
-        inputs=[],
-        outputs=prompt_input
-    )
-    clothes_change_btn1.click(
-        fn=lambda: "(#1의 여성모겔)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    clothes_change_btn2.click(
-        fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)를 직접 모델이 착용한 것처럼 (#3의 장소)에서 의자에 앉아 있는 자연스러운 이미지를 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    holding_product_btn.click(
-        fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 와인잔)을 여성모델이 홍보할 와인잔을 돋보이게 들고 있는 자연스러운 모습으로 이미지를 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    background_change_btn.click(
-        fn=lambda: "(#1의 여성모델)이 (#2 이미지의 배경)을 주요 피사체는 그대로 유지하여 두 이미지의 분위기가 자연스럽게 어우러지도록 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    composite_product_btn.click(
-        fn=lambda: "(#1의 레고모형)에서 청색상어레고를 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우러지도록 채워주세요. 단, 이미지의 다른 부분의 주요 요소는 동일하게 유지해 해야한다.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    
-    submit_btn.click(
-        fn=generate_multiple_images,
-        inputs=[image1_input, image2_input, image3_input, prompt_input],
-        outputs=[output_image1, output_image2, output_image3, output_image4, output_text, prompt_display],
-    )
+            upscale_btn.click(
+                fn=upscale_image,
+                inputs=[input_upscale_image],
+                outputs=[output_upscale_image]
+            )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)