Update app.py

app.py

@@ -2,9 +2,9 @@ import streamlit as st
 from datasets import load_dataset
 from transformers import CLIPProcessor, CLIPModel
 from PIL import Image
-import random
 from io import BytesIO
 import requests
+import random
 
 # Load the CLIP model and processor
 st.title("Meme Battle AI")
@@ -52,14 +52,16 @@ st.write("### Select two memes and let the AI determine the winner!")
 if st.button("Start Meme Battle"):
     meme1, meme2 = fetch_random_memes()
 
-    # Fetch data for Meme 1
-    caption1 = meme1["caption"]
+    # Extract caption and image for Meme 1
+    caption1 = meme1["answers"][0] if meme1["answers"] else "No caption available"
     image_url1 = meme1["image"]
-    score1 = score_meme(image_url1, caption1)
 
-    # Fetch data for Meme 2
-    caption2 = meme2["caption"]
+    # Extract caption and image for Meme 2
+    caption2 = meme2["answers"][0] if meme2["answers"] else "No caption available"
     image_url2 = meme2["image"]
+
+    # Fetch scores
+    score1 = score_meme(image_url1, caption1)
     score2 = score_meme(image_url2, caption2)
 
     # Display Meme 1