src/backend/dataset.py

import pandas as pd
import requests
from PIL import Image

file_name='src/backend/data/images_dataset.csv'

def get_df(start_index,end_index):
    data=pd.read_csv(file_name)
    final_df=data[["photo_id","photo_image_url"]]
    df = final_df[start_index:end_index] 
    return df

def get_image_from_url(url):
    request = requests.get(url,stream=True).raw
    image=Image.open(request)
    return image