Update app.py

app.py

@@ -9,8 +9,8 @@ pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.enable_model_cpu_offload()
 
 def infer(prompt):
-    #prompt = "Darth Vader is surfing on waves"
-    video_frames = pipe(prompt, num_inference_steps=40, height=320, width=576, num_frames=24).frames
+    negative_prompt = "text, watermark, copyright, blurry, nsfw"
+    video_frames = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=40, height=320, width=576, num_frames=24).frames
     video_path = export_to_video(video_frames)
     print(video_path)
     return video_path, gr.Group.update(visible=True)
@@ -78,6 +78,26 @@ img[src*='#center'] {
     display: block;
     margin: auto;
 }
+
+.footer {
+        margin-bottom: 45px;
+        margin-top: 10px;
+        text-align: center;
+        border-bottom: 1px solid #e5e5e5;
+    }
+    .footer>p {
+        font-size: .8rem;
+        display: inline-block;
+        padding: 0 10px;
+        transform: translateY(10px);
+        background: white;
+    }
+    .dark .footer {
+        border-color: #303030;
+    }
+    .dark .footer>p {
+        background: #0b0f19;
+    }
 """
 
 with gr.Blocks(css=css) as demo:
@@ -95,6 +115,7 @@ with gr.Blocks(css=css) as demo:
         )
 
         prompt_in = gr.Textbox(label="Prompt", placeholder="Darth Vader is surfing on waves", elem_id="prompt-in")
+        #neg_prompt = gr.Textbox(label="Negative prompt", value="text, watermark, copyright, blurry, nsfw", elem_id="neg-prompt-in")
         #inference_steps = gr.Slider(label="Inference Steps", minimum=10, maximum=100, step=1, value=40, interactive=False)
         submit_btn = gr.Button("Submit")
         video_result = gr.Video(label="Video Output", elem_id="video-output")
@@ -104,6 +125,25 @@ with gr.Blocks(css=css) as demo:
             loading_icon = gr.HTML(loading_icon_html)
             share_button = gr.Button("Share to community", elem_id="share-btn")
 
+        gr.HTML("""
+            <div class="footer">
+                <p>
+                <a href="https://huggingface.co/cerspense/zeroscope_v2_576w" target="_blank">Zeroscope v2 576w</a> by @cerspence - 
+                Demo by 🤗 <a href="https://twitter.com/fffiloni" target="_blank">Sylvain Filoni</a>
+                </p>
+            </div>
+            <div id="may-like-container" style="display: flex;justify-content: center;flex-direction: column;align-items: center;">
+                <p style="font-size: 0.8em;margin-bottom: 4px;">You may also like: </p>
+                <div id="may-like" style="display:flex; align-items:center; justify-content: center;height:20px;">
+                    <svg height="20" width="148" style="margin-left:4px">       
+                        <a href="https://huggingface.co/spaces/fffiloni/zeroscope-XL" target="_blank">
+                            <image href="https://img.shields.io/badge/🤗 Spaces-Zeroscope XL-blue" src="https://img.shields.io/badge/🤗 Spaces-Image to Music-blue.png" height="20"/>
+                        </a>
+                    </svg>
+                </div>
+            </div>
+        """)
+        
     submit_btn.click(fn=infer,
                     inputs=[prompt_in],
                     outputs=[video_result, share_group])