Update app.py

app.py

@@ -6,7 +6,7 @@ import spaces
 from OmniGen import OmniGenPipeline
 
 pipe = OmniGenPipeline.from_pretrained(
-    "shitao/tmp-preview"
+    "Shitao/OmniGen-v1"
 )
 
 @spaces.GPU(duratio=120)
@@ -46,6 +46,17 @@ def generate_image(text, img1, img2, img3, height, width, guidance_scale, infere
 
 def get_example():
     case = [
+        [
+            "A woman holds a bouquet of flowers and faces the camera.",
+            None,
+            None,
+            None,
+            1024,
+            1024,
+            3.0,
+            50,
+            42,
+        ],
         [
             "A woman holds a bouquet of flowers and faces the camera. Thw woman is the one in <img><|image_1|></img>.",
             "./imgs/test_cases/liuyifei.png",
@@ -77,7 +88,7 @@ def run_for_examples(text, img1, img2, img3, height, width, guidance_scale, infe
 
 # Gradio 接口
 with gr.Blocks() as demo:
-    gr.Markdown("# OmniGen: Unified Image Generation")
+    gr.Markdown("# OmniGen: Unified Image Generation [paper](https://arxiv.org/abs/2409.11340) [code](https://github.com/VectorSpaceLab/OmniGen)")
     with gr.Row():
         with gr.Column():
             # 文本输入框