Update app.py

app.py

@@ -12,16 +12,14 @@ def query1():
     response = requests.get(API_URL1)
     #response = requests.get(API_URL1, headers=headers)
     return response.json()
-    data = query1()
-    print (data)
 
 
-def query2(fetch_url,offset=0,length=10):
+def query2(fetch_url,config="TEST",split="test",offset=0,length=10):
     if fetch_url=="":
         fetch_url = "nlphuji/flickr30k"
     #offset=0
     #length=10
-    API_URL2 = f"https://datasets-server.huggingface.co/rows?dataset={fetch_url}&config=TEST&split=test&offset={offset}&length={length}"
+    API_URL2 = f"https://datasets-server.huggingface.co/rows?dataset={fetch_url}&config={config}&split={split}&offset={offset}&length={length}"
     response = requests.get(API_URL2)
     dictionary=response.json()
     return dictionary
@@ -50,5 +48,5 @@ with gr.Blocks() as app:
             out_find = gr.JSON()
 
     find_btn.click(find_fn,[find_string,out_json],[out_find,out_gal])
-    fetch_btn.click(query2,data_set_url,out_json)
+    fetch_btn.click(query1,data_set_url,out_json)
 app.launch()