Update pages/Visualization.py

pages/Visualization.py

@@ -31,21 +31,16 @@ image_idea = st.text_input('Предложите свою тему для ген
 image_gen__btn = st.button('Генерировать изображение')
 if image_gen__btn:
     with st.spinner('Идёт загрузка изображения...'):
-        try:
-            image_bytes = generate_img({"inputs": image_idea}) 
-            image_raw = io.BytesIO(image_bytes) 
-            st.success('Готово')            
-            st.image(image_raw)     
-            st.markdown('## Опишите фотографию на английском языке') 
-            st.markdown('## План ответа поможет вам:') 
-            st.markdown('+ the place;') 
-            st.markdown('+ the action;') 
-            st.markdown('+ the person’s appearance;') 
-            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="⚠️")
-           
-
+        image_bytes = generate_img({"inputs": image_idea}) 
+        image_raw = io.BytesIO(image_bytes) 
+        st.success('Готово')            
+        st.image(image_raw)     
+        st.markdown('## Опишите фотографию на английском языке') 
+        st.markdown('## План ответа поможет вам:') 
+        st.markdown('+ the place;') 
+        st.markdown('+ the action;') 
+        st.markdown('+ the person’s appearance;') 
+        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 …” ')
 st.divider()