Update app.py

app.py

@@ -30,11 +30,12 @@ class Model(nn.Module):
     def forward(self, x):
         new_tensor = torch.randn(1, 1)
         self.buffer = torch.cat([self.buffer, new_tensor], dim=0)
+        return self
 
-def dict2model(dict,model=Model()):
-    model(dict)
+def dict2model(dict):
+    m = model(dict)
     mix = ModelMixin()
-    mix(model)
+    mix(m)
     return mix
 
 def forest_schnell():