Update app.py

app.py

@@ -7,8 +7,8 @@ from datasets import load_dataset
 #print (dataset)
 
 #headers = {"Authorization": f"Bearer {API_TOKEN}"}
-API_URL1 = "https://datasets-server.huggingface.co/splits?dataset=nlphuji/flickr30k"
 def query1():
+    API_URL1 = "https://datasets-server.huggingface.co/splits?dataset=nlphuji/flickr30k"
     response = requests.get(API_URL1)
     #response = requests.get(API_URL1, headers=headers)
     return response.json()
@@ -16,12 +16,12 @@ def query1():
     print (data)
 
 
-def query2(fetch_url):
+def query2(fetch_url,offset=0,length=10):
     if fetch_url=="":
         fetch_url = "nlphuji/flickr30k"
-    
-    API_URL2 = f"https://datasets-server.huggingface.co/rows?dataset={fetch_url}&config=TEST&split=test&offset=150&length=10"
-    
+    #offset=0
+    #length=10
+    API_URL2 = f"https://datasets-server.huggingface.co/rows?dataset={fetch_url}&config=TEST&split=test&offset={offset}&length={length}"
     response = requests.get(API_URL2)
     dictionary=response.json()
     return dictionary
@@ -30,6 +30,7 @@ def find_fn(inp,out_json):
     print(out_json['rows'])
     print (inp)
     img_list=[]
+    out_json=(query2("",offset=10,lenght=20)
     for rows in out_json['rows']:
         img_ea = rows['row']['image']['src']
         img_list.append(img_ea)