Update app.py

app.py

@@ -32,11 +32,12 @@ if uploaded_file is not None:
     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)
+        
         
     button = st.button('Сгенерировать описание')
-    if button:        
+    if button:  
+        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]