Update app.py

app.py

@@ -207,17 +207,22 @@ if option == "Auto-Regressor":
             
         my_bar.progress(100)
         time.sleep(1)
+
         
-        counter = 0
-        for i in range(4):
-            figout = plt.figure(figsize=(9,6))
-            for j in range(9):
-                plt.subplot(990 + 1 + j)
-                plt.imshow(pixels[counter,:,:,0])#, cmap='gray_r')
-                counter += 1
-                plt.axis('off')
-            plt.show()
-            st.pyplot(figout)
+        from PIL import Image
+        # figout = plt.figure(figsize=(9,6))
+        st.pyplot(Image.fromarray((pixels[-1] * 255).astype(np.uint8), 'RGB').show())
+        
+        # counter = 0
+        # for i in range(4):
+        #     figout = plt.figure(figsize=(9,6))
+        #     for j in range(9):
+        #         plt.subplot(990 + 1 + j)
+        #         plt.imshow(pixels[counter,:,:,0])#, cmap='gray_r')
+        #         counter += 1
+        #         plt.axis('off')
+        #     plt.show()
+        #     st.pyplot(figout)
         
         # %%
 # else: