added progress bar

app.py

@@ -13,7 +13,12 @@ bw = st.number_input('Scale',value=3.05)
 def compute():
     api = nflow(dim=8,latent=16,dataset=uploaded_file)
     api.compile(optim=torch.optim.ASGD,bw=bw,lr=0.0001,wd=None)
-    api.train(iters=10000)
+    
+    my_bar = st.progress(0, text='Currently in progress')
+    
+    for idx in api.train(iters=10000):
+        my_bar.progress(idx[0]/10000, text=str(idx[1]))
+    
     samples = np.array(api.model.sample(
             torch.tensor(api.scaled).float()).detach())
 

normflows.py

@@ -341,8 +341,7 @@ class nflow():
 
             if idx % 100 == 0:
                 print("Loss {}".format(loss.item()))
-                
-        plt.plot(self.losses)
+                yield idx,loss.item()
         
     def performance(self):
         """