Update app.py

app.py

@@ -49,9 +49,9 @@ pipe = QwenImagePipeline.from_pretrained(
 
 
 
-speedup = hf_hub_download(repo_id="witcherderivia/Qwen-Image-Style-Transfer", filename="diffsynth_Qwen-Image-Edit-2509-Lightning-4steps-V1.0-bf16.safetensors")
-qwenstyle= hf_hub_download(repo_id="witcherderivia/Qwen-Image-Style-Transfer", filename="diffusers_Qwen-Image-Edit-2509-Style-Transfer-V1.safetensors")
 
+qwenstyle= hf_hub_download(repo_id="Tele-AI/TeleStyleV2", filename="diffusers-TeleStyleV2-QIE-2509-Lora-bf16.safetensors")
+speedup = hf_hub_download(repo_id="Tele-AI/TeleStyleV2", filename="QIE-2509-Lightning-4steps-V1.0-bf16.safetensors")
 
 
 
@@ -105,7 +105,7 @@ def infer(
 
     
     content_text_input='describe main objects (fewer than 3) with separated words, each word is separated by comma,  the total number of words is strictly fewer than 3'
-    style_text_input='describe only the artistic style, material and stroke in 5 words, not objects.'
+    style_text_input='describe only the artistic style, material and stroke, lighting, color in 5 words, not objects.'
     #pipe.text_encoder.eval()
     content_prompt=''
     style_prompt=''
@@ -385,7 +385,18 @@ with gr.Blocks() as demo:
             gr.Markdown("changing the minedge could lead to different style similarity.")
             default_prompt='Style Transfer the style of Figure 2 to Figure 1, and keep the content and characteristics of Figure 1.'
             gr.Examples(examples=[
+                ['./qwenstyleref/content_1.webp','./qwenstyleref/style_1.jpg',default_prompt,123,False,1.0,4,1024,[]],
+                ['./qwenstyleref/content_6.jpg','./qwenstyleref/style_6.png',default_prompt,123,False,1.0,4,1024,[]],
+                ['./qwenstyleref/style_6.png','./qwenstyleref/content_6.jpg','',123,False,1.0,4,1024,["infer with style prompt"]],
+                ['./qwenstyleref/content_3.png','./qwenstyleref/style_3.png',default_prompt,123,False,1.0,4,1024,[]],
+                ['./qwenstyleref/content_4.png','./qwenstyleref/style_4.png',default_prompt,123,False,1.0,4,1024,[]],
+                ['./qwenstyleref/content_9.jpg','./qwenstyleref/style_9.png',default_prompt,123,False,1.0,4,1024,[]],
+                ['./qwenstyleref/content_11.png','./qwenstyleref/style_11.jpg',default_prompt,123,False,1.0,4,832,[]],
+                ['./qwenstyleref/content_7.png','./qwenstyleref/content_4.png',default_prompt,123,False,1.0,4,1024,[]],
+                '''
+            
                 ['./qwenstyleref/pulpfiction_2.jpg','./qwenstyleref/styleref=6_style_ref.png',default_prompt,123,False,1.0,4,832],
+                
                 ['./qwenstyleref/styleref=0_content_ref.png','./qwenstyleref/110.png',default_prompt,123,False,1.0,4,832],
                 ['./qwenstyleref/romanholiday_1.jpg','./qwenstyleref/s0099____1113_01_query_1_img_000146_1682705733350_08158389675901344.jpg.jpg',default_prompt,123,False,1.0,4,800],
                 ['./qwenstyleref/styleref=0_content_ref.png','./qwenstyleref/125.png',default_prompt,123,False,1.0,4,832],
@@ -397,6 +408,7 @@ with gr.Blocks() as demo:
                 ['./qwenstyleref/styleref=0_content_ref.png','./qwenstyleref/Jotarokujo.webp',default_prompt,123,False,1.0,4,832],
                 ['./qwenstyleref/wallstreet1.jpg','./qwenstyleref/034.png',default_prompt,123,False,1.0,4,1024],
                 ['./qwenstyleref/bird.jpeg','./qwenstyleref/styleref=s0539.png',default_prompt,123,False,1.0,4,832],
+                '''
                 
 
                 
@@ -409,6 +421,7 @@ with gr.Blocks() as demo:
                     true_guidance_scale,
                     num_inference_steps,
                     minedge,
+                    checkbox
                     ], 
                 outputs=[result, seed, content_prompt, style_prompt,prompt], 
                 fn=infer,