Update app.py

app.py

@@ -22,21 +22,21 @@ num_beams = 7
 gen_kwargs = {"max_length": max_length, "min_length": min_length, "num_beams": num_beams}
 
 
-
-uploaded_file = st.file_uploader("Выберите изображение обложки книги в формате jpeg или jpg...", type=["jpg", "jpeg"])
+st.subheader('')
+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")
-        
-    
-    if button:
         pixel_values = feature_extractor(images=[image], return_tensors="pt").pixel_values
         pixel_values = pixel_values.to(device)
+        
+    button = st.button('Сгенерировать описание')
+    if button:        
         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]