change

README.md

@@ -10,4 +10,4 @@ pinned: false
 short_description: CogView4-6B supports Control features, allowing for multiple
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+## CogView4-Control

app.py

@@ -163,11 +163,9 @@ def resize_image_to_valid_dimensions(image, min_size=512, max_size=2048, divisib
     return resized_image
 
 
-# Function to display image information
 def show_image_info(image):
-    """Display information about the uploaded image and its resized dimensions"""
     if image is None:
-        return "未上传图像"
+        return "Must Upload a Image"
 
     if not isinstance(image, Image.Image):
         image = Image.fromarray(image)
@@ -176,7 +174,7 @@ def show_image_info(image):
     resized_img = resize_image_to_valid_dimensions(image)
     new_width, new_height = resized_img.size
 
-    return f"原始尺寸: {orig_width}x{orig_height}, 调整后: {new_width}x{new_height}"
+    return f"Original Size: {orig_width}x{orig_height}, Resize: {new_width}x{new_height}"
 
 
 def infer(prompt, control_image, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, num_images,
@@ -184,18 +182,14 @@ def infer(prompt, control_image, seed, randomize_seed, width, height, guidance_s
     if randomize_seed:
         seed = random.randint(0, 65536)
     if control_image is None:
-        return None, seed, "错误: 请上传控制图像"
+        return None, seed, "Must Upload an Image"
     if not isinstance(control_image, Image.Image):
         control_image = Image.fromarray(control_image)
 
-    try:
+    control_image = resize_image_to_valid_dimensions(control_image)
+    actual_width, actual_height = control_image.size
+    print(f"Change Image Size to: {actual_width}x{actual_height}")
 
-        control_image = resize_image_to_valid_dimensions(control_image)
-        actual_width, actual_height = control_image.size
-        print(f"控制图像尺寸调整为: {actual_width}x{actual_height}")
-    except Exception as e:
-        print(f"图像处理错误: {e}")
-        return None, seed, f"图像处理错误: {e}"
 
     images = pipe(
         prompt=prompt,
@@ -242,15 +236,15 @@ with gr.Blocks() as demo:
     with gr.Column():
         with gr.Row():
             prompt = gr.Text(
-                label="提示词（Prompt）",
+                label="Prompt",
                 show_label=True,
                 max_lines=15,
-                placeholder="输入您的提示词",
+                placeholder="Enter your prompt",
                 container=True,
             )
         with gr.Row():
-            enhance = gr.Button("增强提示词 (强烈建议)", scale=1)
-            run_button = gr.Button("运行", scale=1)
+            enhance = gr.Button("Enhance Prompt (Strongly Suggest)", scale=1)
+            run_button = gr.Button("Run", scale=1)
             enhance.click(
                 convert_prompt,
                 inputs=[prompt],
@@ -274,13 +268,13 @@ with gr.Blocks() as demo:
                 outputs=[image_info]
             )
         with gr.Column(scale=1):
-            result = gr.Gallery(label="生成结果", show_label=True, columns=2, rows=2)
-            error_output = gr.Text(label="错误信息", visible=False)
+            result = gr.Gallery(label="Results", show_label=True, columns=2, rows=2)
+            error_output = gr.Text(label="Errors", visible=False)
 
     with gr.Row():
         num_images = gr.Radio(
             choices=[1, 2, 4],
-            label="生成图像数量",
+            label="Number of Images",
             value=1
         )
 
@@ -295,43 +289,38 @@ with gr.Blocks() as demo:
         return gr.update(maximum=max_width)
 
 
-    with gr.Accordion("高级设置", open=False):
+    with gr.Accordion("Advanced Settings", open=False):
         seed = gr.Slider(
-            label="种子值",
+            label="Seed",
             minimum=0,
             maximum=65536,
             step=1,
             value=0,
         )
 
-        randomize_seed = gr.Checkbox(label="随机种子", value=True)
+        randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
         with gr.Row():
             width = gr.Slider(
-                label="宽度",
+                label="Width",
                 minimum=512,
                 maximum=2048,
                 step=32,
                 value=1024,
-                interactive=False,
-                info="宽度将根据上传的控制图像自动调整"
             )
 
             height = gr.Slider(
-                label="高度",
+                label="Height",
                 minimum=512,
                 maximum=2048,
                 step=32,
                 value=1024,
-                interactive=False,
-                info="高度将根据上传的控制图像自动调整"
             )
 
-        gr.Markdown("**注意**：图像尺寸会自动根据控制图像调整，保持比例，确保宽高都能被32整除，并且不超过2^21像素总量。")
-
+        gr.Markdown("Warning: The width and height must be divisible by 32 and Image size must not exceed 2^21 pixels.")
         with gr.Row():
             guidance_scale = gr.Slider(
-                label="引导尺度（Guidance scale）",
+                label="Guidance scale",
                 minimum=0.0,
                 maximum=10.0,
                 step=0.1,
@@ -339,7 +328,7 @@ with gr.Blocks() as demo:
             )
 
             num_inference_steps = gr.Slider(
-                label="推理步数",
+                label="Number of inference steps",
                 minimum=10,
                 maximum=100,
                 step=1,
@@ -349,7 +338,6 @@ with gr.Blocks() as demo:
         gr.Markdown("### Examples")
         for i, pair in enumerate(example_pairs):
             with gr.Row():
-                # Input image
                 ex_input_img = gr.Image(
                     value=pair["input"],
                     label="Input",

requirements.txt

@@ -1,6 +1,6 @@
 accelerate>=1.3.0
 sentencepiece>=0.2.0
-git+https://github.com/huggingface/diffusers.git
+git+https://github.com/zRzRzRzRzRzRzR/diffusers.git@cogview4_control
 gradio>=5.15.0
 transformers>=4.48.3
 openai>=1.61.0