Update pages/Visualization.py

pages/Visualization.py

@@ -26,7 +26,6 @@ if image_gen__btn:
         try:
             image_bytes = generate_img({"inputs": image_idea}) 
             image_raw = io.BytesIO(image_bytes) 
-        #except UnidentifiedImageError:            
             st.success('Готово')            
             st.image(image_raw)     
             st.markdown('## Опишите фотографию на английском языке') 
@@ -37,6 +36,9 @@ if image_gen__btn:
             st.markdown('+ whether you like the picture or not;') 
             st.markdown('+ why.') 
             st.markdown('Start with: “I’d like to describe this picture. The picture shows …” ')
+        except UnidentifiedImageError:  
+            st.warning('This is a warning', icon="⚠️")
+           
 
 st.divider()