Update app.py

app.py

@@ -33,9 +33,10 @@ if uploaded_file is not None:
     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)
+    
     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]