Upload 45 files

README.md

@@ -1,6 +1,6 @@
 ---
 title: AnyStory
-emoji: 💻
+emoji: ✏️
 colorFrom: indigo
 colorTo: yellow
 sdk: gradio

app.py

@@ -10,7 +10,6 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 
-from src.agent import extend_prompt, translate_prompt
 from src.anystory import call_anystory
 from src.matting import ImageUniversalMatting
 from src.util import upload_pil_2_oss
@@ -112,8 +111,6 @@ def interface():
 
             with gr.Group():
                 prompt = gr.Textbox(value="", label='Prompt', lines=6, show_label=True)
-                en_prompt = gr.Textbox(value="", label='prompt', lines=6, show_label=True, visible=False)
-                # prompt_extend_button = gr.Button(value="提示词扩写")
 
         with gr.Column(scale=1, min_width=100):
             result_gallery = gr.Image(type="pil", label="Generated Image", visible=True, height=450)
@@ -139,23 +136,11 @@ def interface():
         fn=set_image_seg_finished, outputs=generated_information
     )
 
-    # prompt_extend_button.click(
-    #     fn=set_prompt_extend_unfinished, outputs=generated_information
-    # ).then(
-    #     fn=extend_prompt, inputs=[prompt], outputs=[prompt]
-    # ).then(
-    #     fn=set_prompt_extend_finished, outputs=generated_information
-    # )
-
     run_button.click(
-        fn=set_prompt_translate_unfinished, outputs=generated_information
-    ).then(
-        fn=translate_prompt, inputs=[prompt], outputs=[en_prompt]
-    ).then(
         fn=set_image_generate_unfinished, outputs=generated_information
     ).then(
         fn=process,
-        inputs=[pil_subject_A_image, pil_subject_A_mask, pil_subject_B_image, pil_subject_B_mask, en_prompt],
+        inputs=[pil_subject_A_image, pil_subject_A_mask, pil_subject_B_image, pil_subject_B_mask, prompt],
         outputs=[result_gallery]
     ).then(
         fn=set_image_generate_finished, outputs=generated_information
@@ -247,7 +232,8 @@ def interface():
         gr.Examples(
             label="Examples",
             examples=examples,
-            inputs=[pil_subject_A_image, pil_subject_A_mask, pil_subject_B_image, pil_subject_B_mask, prompt, result_gallery],
+            inputs=[pil_subject_A_image, pil_subject_A_mask, pil_subject_B_image, pil_subject_B_mask, prompt,
+                    result_gallery],
         )
 
 
@@ -262,24 +248,6 @@ def set_image_seg_finished():
     return gr.update(visible=True, value="<h3>Subject mask ready!</h3>")
 
 
-def set_prompt_extend_unfinished():
-    return gr.update(
-        visible=True,
-        value="<h3>(Unfinished) Rewriting your prompt... ✍️</h3>",
-    )
-
-
-def set_prompt_extend_finished():
-    return gr.update(visible=True, value="<h3>Prompt expanded successfully!</h3>")
-
-
-def set_prompt_translate_unfinished():
-    return gr.update(
-        visible=True,
-        value="<h3>(Unfinished) Preprocessing...</h3>",
-    )
-
-
 def set_image_generate_unfinished():
     return gr.update(
         visible=True,
@@ -301,7 +269,7 @@ if __name__ == "__main__":
                     &nbsp;
                     <a href='https://aigcdesigngroup.github.io/AnyStory/'><img src='https://img.shields.io/badge/Project_Page-AnyStory-green' alt='Project Page'></a>
                     &nbsp;
-                    <a href='https://modelscope.cn/studios/damo/studio_anytext'><img src='https://img.shields.io/badge/Demo-ModelScope-blue'></a>
+                    <a href='https://modelscope.cn/studios/iic/AnyStory'><img src='https://img.shields.io/badge/Demo-ModelScope-blue'></a>
                 </div>
                 </br>
             </div>
@@ -313,7 +281,7 @@ if __name__ == "__main__":
         """
 
     description = r"""🚀🚀🚀 Quick Start:<br>
-        1. Upload subject reference images (clean background; real human IDs unsupported for now), Add prompts (CN/EN supported), and Click "<b>RUN</b>".<br>
+        1. Upload subject reference images (clean background; real human IDs unsupported for now), Add prompts (only EN supported), and Click "<b>RUN</b>".<br>
         2. (Recommended) Click "<b>Segment Subject</b>" to create masks (or upload your own B&W masks) for subjects. This helps the model better reference the subject you specify (otherwise, we will perform automatic detection). 🤗<br>
         """
 

requirements.txt

@@ -2,3 +2,4 @@ oss2
 phidata
 dashscope
 tensorflow==2.15
+openai