update

app.py

@@ -14,8 +14,10 @@ elif torch.cuda.is_available():
 else:
     DEVICE = "cpu"
 
-edit_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl_edit.safetensors")
-normal_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl.safetensors")
+#edit_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl_edit.safetensors")
+#normal_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl.safetensors")
+edit_file = hf_hub_download(repo_id="cocktailpeanut/c", filename="cosxl_edit.safetensors")
+normal_file = hf_hub_download(repo_id="cocktailpeanut/c", filename="cosxl.safetensors")
 
 def set_timesteps_patched(self, num_inference_steps: int, device = None):
     self.num_inference_steps = num_inference_steps
@@ -64,16 +66,6 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown('''# CosXL demo
     Unofficial demo for CosXL, a SDXL model tuned to produce full color range images. CosXL Edit allows you to perform edits on images. Both have a [non-commercial community license](https://huggingface.co/stabilityai/cosxl/blob/main/LICENSE)
     ''')
-    with gr.Tab("CosXL"):
-      with gr.Group():
-          with gr.Row():
-            prompt_normal = gr.Textbox(show_label=False, scale=4, placeholder="Your prompt, e.g.: backlit photography of a dog")
-            button_normal = gr.Button("Generate", min_width=120)
-          output_normal = gr.Image(label="Your result image", interactive=False)
-          with gr.Accordion("Advanced Settings", open=False):
-            negative_prompt_normal = gr.Textbox(label="Negative Prompt")
-            guidance_scale_normal = gr.Number(label="Guidance Scale", value=7)
-      gr.Examples(examples=normal_examples, fn=run_normal, inputs=[prompt_normal], outputs=[output_normal], cache_examples=True) 
     with gr.Tab("CosXL Edit"):
       with gr.Group():
           image_edit = gr.Image(label="Image you would like to edit", type="pil")
@@ -84,7 +76,17 @@ with gr.Blocks(css=css) as demo:
           with gr.Accordion("Advanced Settings", open=False):
             negative_prompt_edit = gr.Textbox(label="Negative Prompt")
             guidance_scale_edit = gr.Number(label="Guidance Scale", value=7)
-      gr.Examples(examples=edit_examples, fn=run_edit, inputs=[image_edit, prompt_edit], outputs=[output_edit], cache_examples=True)
+      gr.Examples(examples=edit_examples, fn=run_edit, inputs=[image_edit, prompt_edit], outputs=[output_edit], cache_examples=False)
+    with gr.Tab("CosXL"):
+      with gr.Group():
+          with gr.Row():
+            prompt_normal = gr.Textbox(show_label=False, scale=4, placeholder="Your prompt, e.g.: backlit photography of a dog")
+            button_normal = gr.Button("Generate", min_width=120)
+          output_normal = gr.Image(label="Your result image", interactive=False)
+          with gr.Accordion("Advanced Settings", open=False):
+            negative_prompt_normal = gr.Textbox(label="Negative Prompt")
+            guidance_scale_normal = gr.Number(label="Guidance Scale", value=7)
+      gr.Examples(examples=normal_examples, fn=run_normal, inputs=[prompt_normal], outputs=[output_normal], cache_examples=False) 
     button_edit.click(
         
     )