Update app.py

app.py

@@ -27,7 +27,7 @@ class MyModel(nn.Module):
         super().__init__()
         self.register_buffer('a', torch.ones(1, 1))
 
-    def forward(self, x: torch.Tensor, extend: bool):
+    def forward(self, x, extend=False):
         if extend:
             new_tensor = torch.randn(1, 1)
             self.a = torch.cat([self.a, new_tensor], dim=0)