Re-add images under Git LFS control

app.py

@@ -13,7 +13,7 @@ from openai import OpenAI
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = CogView4Pipeline.from_pretrained("ZP2HF/CogView4-6B-0125", torch_dtype=torch.bfloat16).to(device)
+pipe = CogView4Pipeline.from_pretrained("/share/zyx/CogView4-6B-0125/", torch_dtype=torch.bfloat16).to(device)
 
 
 def clean_string(s):
@@ -96,8 +96,8 @@ def delete_old_files():
     while True:
         now = datetime.now()
         cutoff = now - timedelta(minutes=5)
+        os.makedirs("./gradio_tmp", exist_ok=True)
         directories = ["./gradio_tmp"]
-
         for directory in directories:
             for filename in os.listdir(directory):
                 file_path = os.path.join(directory, filename)
@@ -111,7 +111,7 @@ def delete_old_files():
 threading.Thread(target=delete_old_files, daemon=True).start()
 
 
-@spaces.GPU  # [uncomment to use ZeroGPU]
+@spaces.GPU(duration=120)  # [uncomment to use ZeroGPU]
 def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
           progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
@@ -130,16 +130,16 @@ def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_infer
 
 
 examples = [
-    "A vintage pink convertible with glossy chrome finishes and whitewall tires sits parked on an open road, surrounded by a field of wildflowers under a clear blue sky. The car's body is a delicate pastel pink, complementing the vibrant greens and colors of the meadow. Its interior boasts cream leather seats and a polished wooden dashboard, evoking a sense of classic elegance. The sun casts a soft light on the vehicle, highlighting its curves and shiny surfaces, creating a picture of nostalgia mixed with dreamy escapism.",
-    "A noble black Labrador retriever sits serenely in a sunlit meadow, its glossy coat absorbing the golden rays of a late afternoon sun. The dog's intelligent eyes sparkle with a mixture of curiosity and loyalty, as it gazes off into the distance where the meadow meets a line of tall, slender birch trees. The dog's posture is regal, yet approachable, with its tongue playfully hanging out to the side slightly, suggesting a friendly disposition. The idyllic setting is filled with the vibrant greens of lush grass and the soft colors of wildflowers speckled throughout, creating a peaceful harmony between the dog and its natural surroundings.",
-    "A vibrant red-colored dog of medium build stands attentively in an autumn forest setting. Its fur is a deep, rich red, reminiscent of autumn leaves, contrasting with its bright, intelligent eyes, a clear sky blue. The dog's ears perk up, and its tail wags slightly as it looks off into the distance, its posture suggesting alertness and curiosity. Golden sunlight filters through the canopy of russet and gold leaves above, casting dappled light onto the forest floor and the glossy coat of the canine, creating a serene and heartwarming scene."
+    "A surreal scene unfolds where a chestnut horse adorned with a flowing crimson cape is perched upon the back of a stoic astronaut. The astronaut, in a gleaming white space suit with reflective visor, stands against a backdrop of a swirling cosmic sky filled with distant stars and nebulae. The horse, with its mane fluttering softly in the non-existent breeze of space, seems to be in command, navigating through the galaxy. The juxtaposition of an equine figure with celestial exploration strikes a whimsical yet grand tone, a vibrant clash of eras and realms.",
+    "A charming panda, adorned with a white apron and a barista's hat, is elegantly pouring steamed milk into a ceramic cup filled with rich, dark espresso. The resulting latte art is a delicate bamboo leaf pattern, perfectly contrasting the creamy white froth and the coffee's deep color. The scene is warmly lit, emanating a cozy café atmosphere, with bamboo décor in the background and the subtle steam rising from the freshly crafted beverage.",
+    "A vintage red convertible with gleaming chrome finishes sits attractively under the golden hues of a setting sun, parked on a deserted cobblestone street in a charming old town. The car's polished body reflects the surrounding quaint buildings and the few early evening stars beginning to twinkle in the gentle gradient of the twilight sky. A light breeze teases the few fallen leaves near the car's pristine white-walled tires, which rest casually by the sidewalk, hinting at the leisurely pace of life in this serene setting."
 ]
 
 
 with gr.Blocks() as demo:
     gr.Markdown("""
             <div style="text-align: center; font-size: 32px; font-weight: bold; margin-bottom: 20px;">
-             CogView4-6B Huggingface Space🤗
+             CogView4-6B Hugging Face Space🤗
            </div>
            <div style="text-align: center;">
                <a href="https://huggingface.co/THUDM/CogView4-6B">🤗 Model Hub | 
@@ -220,10 +220,25 @@ with gr.Blocks() as demo:
                     value=50,
                 )
 
-        gr.Examples(
-            examples=examples,
-            inputs=[prompt]
-        )
+        with gr.Column():
+            gr.Markdown("### Examples")
+            for i, ex in enumerate(examples):
+                with gr.Row():
+                    ex_btn = gr.Button(
+                        value=ex,
+                        variant="secondary",
+                        elem_id=f"ex_btn_{i}",
+                        scale=3
+                    )
+                    ex_img = gr.Image(
+                        value=f"img/img_{i + 1}.png",
+                        label="Effect",
+                        interactive=False,
+                        height=130,
+                        width=130,
+                        scale=1
+                    )
+                    ex_btn.click(fn=lambda ex=ex: ex, inputs=[], outputs=prompt)
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,