Update app.py

app.py

@@ -1,3 +1,4 @@
+import gradio as gr
 import requests
 from datasets import load_dataset
 
@@ -10,14 +11,23 @@ def query1():
     response = requests.get(API_URL1)
     #response = requests.get(API_URL1, headers=headers)
     return response.json()
-data = query1()
-print (data)
+    data = query1()
+    print (data)
 
 
-API_URL2 = "https://datasets-server.huggingface.co/rows?dataset=nlphuji/flickr30k&config=TEST&split=test&offset=150&length=10"
-def query2():
+def query2(fetch_url):
+    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"
+    
     response = requests.get(API_URL2)
     return response.json()
-data = query2()
 
-print (data)
+
+with gr.Blocks as app:
+    data_set_url=gr.Textbox():
+    fetch_btn=gr.Button():
+    out_json = gr.JSON():
+    fetch_btn.click(query2,data_set_url,out_json)
+app.launch()