Update app.py

app.py

@@ -97,7 +97,10 @@ if uploaded_file is not None:
         image_bytes = query({
             "inputs": 'A picture without text about: ' + summary[0]['summary_text']
         })
-        image = Image.open(io.BytesIO(image_bytes))
+        image_width = 800  # Set the desired width
+        image_height = 600  # Set the desired height
+        image = Image.open(io.BytesIO(image_bytes)).resize((image_width, image_height))
+
         temp_img_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
         image.save(temp_img_path)