Update app.py

app.py

@@ -16,7 +16,7 @@ bases = {
     "epiCRealism": "emilianJR/epiCRealism"
 }
 step_loaded = None
-base_loaded = "ToonYou"
+base_loaded = "epiCRealism"
 motion_loaded = None
 
 # Ensure model and scheduler are initialized in GPU-enabled function
@@ -69,7 +69,15 @@ def generate_image(prompt, base, motion, step, progress=gr.Progress()):
     def progress_callback(i, t, z):
         progress((i+1, step))
 
-    output = pipe(prompt=prompt, guidance_scale=1.0, num_inference_steps=step, callback=progress_callback, callback_steps=1)
+    output = pipe(
+        prompt=prompt,
+        width=1024,
+        height=576,
+        guidance_scale=1.0,
+        num_inference_steps=step,
+        callback=progress_callback,
+        callback_steps=1
+    )
 
     has_nsfw_concepts = check_nsfw_images([output.frames[0][0]])
     if has_nsfw_concepts[0]:
@@ -137,8 +145,8 @@ with gr.Blocks(css="style.css") as demo:
     video = gr.Video(
         label='AnimateDiff-Lightning',
         autoplay=True,
-        height=512,
-        width=512,
+        height=1024,
+        width=576,
         elem_id="video_output"
     )