Update app.py

app.py

@@ -33,10 +33,6 @@ class MyModel(nn.Module):
             self.a = torch.cat([self.a, new_tensor], dim=0)
         return x
 
-for _ in range(10):
-    out = model(x, extend=True)
-    print(model.state_dict())
-
 def forest_schnell():
     PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to("cuda")
     return PIPE