Update app.py

app.py

@@ -168,7 +168,7 @@ if option == "Auto-Regressor":
         
         # %%
         # from IPython.display import Image, display
-        from tqdm import tqdm_notebook
+        from tqdm import tqdm
         
         
         # Create an empty array of pixels.
@@ -188,7 +188,7 @@ if option == "Auto-Regressor":
         
         
         # Iterate over the pixels because generation has to be done sequentially pixel by pixel.
-        for row in range(rows):
+        for row in tqdm(range(rows)):
             for col in range(cols):
                 for channel in range(channels):
                     time.sleep(0.01)
@@ -200,10 +200,11 @@ if option == "Auto-Regressor":
                     pixels[:, row, col, channel] = tensorflow.math.ceil(
                       probs - tensorflow.random.uniform(probs.shape)
                     )
-            if row<rows/2:
-                my_bar.progress((rows+1)*2)
-            else:
-                my_bar.progress(row+51)
+            my_bar.progress(row*3.125)
+            # if row<rows/2:
+            #     my_bar.progress((rows+1)*2)
+            # else:
+            #     my_bar.progress(row+51)
             
         my_bar.progress(100)
         time.sleep(1)