Update app.py

app.py

@@ -1,10 +1,10 @@
-import subprocess
-subprocess.call("mim install 'mmengine>=0.6.0'", shell=True)
-subprocess.call("mim install 'mmcv>=2.0.0rc4,<2.1.0'", shell=True)
-subprocess.call("min install 'mmdet>=3.0.0,<4.0.0'", shell=True)
-subprocess.call("mim install 'mmyolo'", shell=True)
-subprocess.call("mim install 'mmpose'", shell=True)
-subprocess.call("mim install 'mmpretrain'", shell=True)
+# import subprocess
+# subprocess.call("mim install 'mmengine>=0.6.0'", shell=True)
+# subprocess.call("mim install 'mmcv>=2.0.0rc4,<2.1.0'", shell=True)
+# subprocess.call("min install 'mmdet>=3.0.0,<4.0.0'", shell=True)
+# subprocess.call("mim install 'mmyolo'", shell=True)
+# subprocess.call("mim install 'mmpose'", shell=True)
+# subprocess.call("mim install 'mmpretrain'", shell=True)
 
 import numpy as np
 import gradio as gr
@@ -61,7 +61,7 @@ def get_access_token(refatch=False) -> str:
         return access_token
 
 
-def resize_image(img, max_length=524, min_length=50) -> Tuple[np.ndarray, bool]:
+def resize_image(img, max_length=2048, min_length=50) -> Tuple[np.ndarray, bool]:
     """Ensure that the longest side is shorter than 524px and the shortest side is longer than 50px.
     :param img: 前端传入的图片
     :param max_length: 最长边像素
@@ -196,7 +196,7 @@ def model2_rem(x):
     return new_img, resized_f
 
 
-def model3_ext(x: np.ndarray, num_clusters=12):
+def model3_ext(x: np.ndarray, num_clusters=8):
     """主色调提取
     :param x: 前端传入的图片
     :param num_clusters: 聚类的数量
@@ -329,7 +329,7 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 model3_input = gr.Image(height=400, image_mode='RGBA')
-                model3_slider = gr.Slider(minimum=1, maximum=20, step=1, value=12,
+                model3_slider = gr.Slider(minimum=1, maximum=20, step=1, value=8,
                                           min_width=400, label="聚类数量")
             with gr.Column():
                 model3_output_img = gr.Image(height=100)