Update app.py

app.py

@@ -92,6 +92,7 @@ def sample_then_run(net):
         sample[0, i] = torch.normal(m[i], standev[i], (1,1))
 
     net = "model_"+str(uuid.uuid4())[:4]+".pt"
+    torch.save(sample, net)
 
 
     return net