Update app.py

app.py

@@ -24,20 +24,22 @@ gen_kwargs = {"max_length": max_length, "min_length": min_length, "num_beams": n
 
 
 uploaded_file = st.file_uploader("Выберите изображение обложки книги в формате jpeg или jpg...", type=["jpg", "jpeg"])
-button = st.button('Сгенерировать описание')
-if uploaded_file is not None and button:
-    image = Image.open(uploaded_file)
-    st.image(image, caption='Загруженное изображение')
-    image = image.resize([224,224])
-    if image.mode != "RGB":
-        image = image.convert(mode="RGB")
-        
-    pixel_values = feature_extractor(images=[image], return_tensors="pt").pixel_values
-    pixel_values = pixel_values.to(device)
-
-    output_ids = model.generate(pixel_values, **gen_kwargs)
-
-    preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
-    preds = [pred.strip() for pred in preds]
-    st.write(preds[0])
+
+if uploaded_file is not None:
+    button = st.button('Сгенерировать описание')
+    if button:
+        image = Image.open(uploaded_file)
+        st.image(image, caption='Загруженное изображение')
+        image = image.resize([224,224])
+        if image.mode != "RGB":
+            image = image.convert(mode="RGB")
+            
+        pixel_values = feature_extractor(images=[image], return_tensors="pt").pixel_values
+        pixel_values = pixel_values.to(device)
+    
+        output_ids = model.generate(pixel_values, **gen_kwargs)
+    
+        preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
+        preds = [pred.strip() for pred in preds]
+        st.write(preds[0])