import gradio as gr import requests import json from datasets import load_dataset #dataset = load_dataset("nlphuji/flickr30k", split="test[10:20]") #print (dataset) #headers = {"Authorization": f"Bearer {API_TOKEN}"} API_URL1 = "https://datasets-server.huggingface.co/splits?dataset=nlphuji/flickr30k" 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): 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) dictionary=response.json() return dictionary def find_fn(inp,out_json): print(out_json['rows']) print (inp) img_list=[] for rows in out_json['rows']: img_ea = rows['row']['image']['src'] img_list.append(img_ea) return img_list,img_list with gr.Blocks() as app: with gr.Row(): data_set_url=gr.Textbox() fetch_btn=gr.Button() with gr.Row(): find_string=gr.Textbox() find_btn=gr.Button("Search") with gr.Row(): out_json = gr.JSON() with gr.Column(): out_gal = gr.Gallery() 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) app.launch()