l

app.py

@@ -17,7 +17,8 @@ def compute():
     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]))
+        st.write('Loss:',idx[1])
+        my_bar.progress(idx[0]/10000)
     
     samples = np.array(api.model.sample(
             torch.tensor(api.scaled).float()).detach())