Update src/streamlit_app.py

src/streamlit_app.py

@@ -24,7 +24,7 @@ import streamlit as st
 import torch
 from sentence_transformers import SentenceTransformer, util
 from transformers import CLIPProcessor, CLIPModel
-from datasets import load_dataset
+from datasets import load_dataset, get_dataset_split_names
 from PIL import Image
 import openai