Update pages/img-2-txt.py

pages/img-2-txt.py

@@ -21,14 +21,14 @@ gen_kwargs = {"max_length": max_length, "min_length": min_length, "num_beams": n
 uploaded_file = st.file_uploader("Выберите изображение обложки книги в формате jpeg или jpg...", type=["jpg", "jpeg"])
 
 if uploaded_file is not None:
-    button = st.button('Сгенерировать описание')
+    
     image = Image.open(uploaded_file)
     st.image(image, caption='Загруженное изображение')
     image = image.resize([224,224])
     if image.mode != "RGB":
         image = image.convert(mode="RGB")
         
-    
+    button = st.button('Сгенерировать описание')
     if button:
         pixel_values = feature_extractor(images=[image], return_tensors="pt").pixel_values
         pixel_values = pixel_values.to(device)