Update app.py

app.py

@@ -52,10 +52,10 @@ cap = str(index)
 
 if st.session_state.show:
     caps[true_idx] = f'{true_idx} (TARGET IMAGE)'
-    img = Image.open(io.BytesIO(requests.get(img_urls[index], stream=True).content))
+    img = Image.open(io.BytesIO(requests.get(img_urls[true_idx], stream=True).content))
     img_width, img_height = img.size
     smaller = min(img_width, img_height)
-    images[true_idx] = ImageOps.expand(img_urls[true_idx],border=smaller//8,fill='green')
+    images[true_idx] = ImageOps.expand(img,border=smaller//8,fill='green')
     if true_idx == index:
         cap = f'{true_idx} (TARGET IMAGE)'
 else: