Update for gradio 3.0

README.md

@@ -4,7 +4,7 @@ emoji: 🏢
 colorFrom: blue
 colorTo: gray
 sdk: gradio
-sdk_version: 2.9.4
+sdk_version: 3.0.2
 app_file: app.py
 pinned: false
 ---

app.py

@@ -39,7 +39,6 @@ def parse_args() -> argparse.Namespace:
                         dest='enable_queue',
                         action='store_false')
     parser.add_argument('--allow-flagging', type=str, default='never')
-    parser.add_argument('--allow-screenshot', action='store_true')
     return parser.parse_args()
 
 
@@ -64,10 +63,10 @@ def generate_z(z_dim: int, seed: int, device: torch.device) -> torch.Tensor:
 
 
 @torch.inference_mode()
-def generate_interpolated_images(
-        seed0: int, psi0: float, seed1: int, psi1: float,
-        num_intermediate: int, model: nn.Module,
-        device: torch.device) -> tuple[list[np.ndarray], np.ndarray]:
+def generate_interpolated_images(seed0: int, psi0: float, seed1: int,
+                                 psi1: float, num_intermediate: int,
+                                 model: nn.Module,
+                                 device: torch.device) -> list[np.ndarray]:
     seed0 = int(np.clip(seed0, 0, np.iinfo(np.uint32).max))
     seed1 = int(np.clip(seed1, 0, np.iinfo(np.uint32).max))
 
@@ -88,8 +87,7 @@ def generate_interpolated_images(
             torch.uint8)
         out = out[0].cpu().numpy()
         res.append(out)
-    concatenated = np.hstack(res)
-    return res, concatenated
+    return res
 
 
 def main():
@@ -118,16 +116,11 @@ def main():
                              default=7,
                              label='Number of Intermediate Frames'),
         ],
-        [
-            gr.outputs.Carousel(gr.outputs.Image(type='numpy'),
-                                label='Output Images'),
-            gr.outputs.Image(type='numpy', label='Concatenated'),
-        ],
+        gr.Gallery(type='numpy', label='Output Images'),
         title=TITLE,
         description=DESCRIPTION,
         article=ARTICLE,
         theme=args.theme,
-        allow_screenshot=args.allow_screenshot,
         allow_flagging=args.allow_flagging,
         live=args.live,
     ).launch(