fix app

app.py

@@ -140,10 +140,6 @@ def infer(prompt, output_size="256x256"):
     grid = (grid + 1.0) / 2.0
     grid = (grid * 255).to(torch.uint8).permute(0, 2, 3, 1)
 
-    print(grid.shape)
-    print(torch.max(grid))
-    print(torch.min(grid))
-
     torchvision.io.write_video(
         video_path,
         grid,
@@ -155,11 +151,17 @@ def infer(prompt, output_size="256x256"):
     print(video_path)
     return video_path
 
+# examples = [
+#     ["A chihuahua in astronaut suit floating in space, cinematic lighting, glow effect",],
+#     ["Campfire at night in a snowy forest with starry sky in the background",],
+#     ["A corgi is swimming quickly",],
+#     ["A panda is surfing in the universe",],
+# ]
 examples = [
-    ["A chihuahua in astronaut suit floating in space, cinematic lighting, glow effect",],
-    ["Campfire at night in a snowy forest with starry sky in the background",],
-    ["A corgi is swimming quickly",],
-    ["A panda is surfing in the universe",],
+    "A chihuahua in astronaut suit floating in space, cinematic lighting, glow effect",
+    "Campfire at night in a snowy forest with starry sky in the background",
+    "A corgi is swimming quickly",
+    "A panda is surfing in the universe",
 ]
 
 css = """
@@ -257,7 +259,6 @@ with gr.Blocks(css=css) as demo:
         )
 
         prompt_in = gr.Textbox(label="Prompt", placeholder="A chihuahua in astronaut suit floating in space, cinematic lighting, glow effect")
-        # output_size = gr.Dropdown(["576x1024", "256x256"], label="Output Size", info="576x1024 is watermark-free")
         output_size = gr.Dropdown(["576x1024", "256x256"], value="256x256", label="Output Size", info="576x1024 is watermark-free")
         submit_btn = gr.Button("Generate")
         video_result = gr.Video(label="Video Output")