Update app.py

app.py

@@ -27,10 +27,10 @@ from modules.model_edit import Step1XParams, Step1XEdit
 print("TORCH_CUDA", torch.cuda.is_available())
 
 examples = [
-["meme.jpg", "turn into an illustration in studio ghibli style",("meme.jpg","ghibli_meme.jpg"),],
-["celeb_meme.jpg", "replace the gray blazer with a leather jacket",("celeb_meme.jpg","leather.jpg")],
-["cookie.png", "remove the cookie",("cookie.png","no_cookie.png")],
-["poster_orig.jpg", "replace 'lambs' with 'llamas'",("poster_orig.jpg","poster.jpg")],
+["examples 2/meme.jpg", "turn into an illustration in studio ghibli style",("examples 2/meme.jpg","examples 2/ghibli_meme.jpg"),],
+["examples 2/celeb_meme.jpg", "replace the gray blazer with a leather jacket",("examples 2/celeb_meme.jpg","examples 2/leather.jpg")],
+["examples 2/cookie.png", "remove the cookie",("examples 2/cookie.png","examples 2/no_cookie.png")],
+["examples 2/poster_orig.jpg", "replace 'lambs' with 'llamas'",("examples 2/poster_orig.jpg","examples 2/poster.jpg")],
 ]
 
 def generate_examples(init_image, prompt):
@@ -394,7 +394,7 @@ class ImageGenerator:
 
 
 # 模型仓库ID（如："bert-base-uncased"）
-model_repo = "MoinulwithAI/TextToImageEdit/"
+model_repo = "stepfun-ai/Step1X-Edit"
 # 本地保存路径
 model_path = "./model_weights"
 os.makedirs(model_path, exist_ok=True)
@@ -449,13 +449,13 @@ def inference(prompt, ref_images, seed, size_level):
 with gr.Blocks() as demo:
     gr.Markdown(
         """
-        # TextToImageEdit
+        # Step1X-Edit
         """
     )
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(
-                label="prompt",
+                label="编辑指令 prompt",
                 value='Remove the person from the image.',
             )
             init_image = gr.Image(label="Input Image", type='pil')