Spaces:
Sleeping
Sleeping
File size: 2,383 Bytes
d714d3d 8fa564a f894099 60ed47f 49687e4 ed88b70 49687e4 8d217ca 9265cc1 8d217ca 36b3c9a 62a99f6 36b3c9a c458717 8512e9b 62a99f6 8e0ab44 62a99f6 8512e9b f894099 3c99043 d714d3d 384f0cb 3c99043 f894099 60ed47f 4ec3536 f894099 adee038 c1e08f5 a8a0883 838a98c a8a0883 f6669c0 fab2bcd adee038 f6669c0 adee038 f6669c0 1f375b6 ef02a4a 36b3c9a ef02a4a 36b3c9a 8512e9b 36b3c9a ef02a4a 9bf98ce 2322049 a8a0883 36b3c9a ef02a4a 8512e9b adee038 8512e9b d714d3d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
import gradio as gr
import pandas as pd
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}"}
def query1(fetch_url):
if fetch_url=="":
fetch_url = "nlphuji/flickr30k"
API_URL1 = f"https://datasets-server.huggingface.co/splits?dataset={fetch_url}"
response = requests.get(API_URL1)
json_obj=response.json()
drop_box = []
for obj in json_obj:
print (obj)
out_config = obj[0]
out_split = obj[1]
print (out_config)
print (out_split)
drop_box.append(obj[0])
#df = pd.DataFrame.from_dict(json_obj)
return json_obj
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={config}&split={split}&offset={offset}&length={length}"
response = requests.get(API_URL2)
dictionary=response.json()
return dictionary
def find_fn(inp,fetch_url,config="TEST",split="test",offset=0,length=10):
#print(out_json['rows'])
#print (inp)
img_list=[]
out_json=query2(fetch_url,config,split,offset=10,length=20)
inp=inp.strip("[]")
print(inp)
for ea in out_json[f"{inp}"]:
#img_ea = ea['row']['image']['src']
img_list.append(ea)
return img_list
with gr.Blocks() as app:
with gr.Row():
data_set_url=gr.Textbox(label="Dataset (repo/name)")
fetch_btn=gr.Button()
with gr.Row():
config_set=gr.Textbox(label="Config")
split_set=gr.Textbox(label="Split")
#config_drop = gr.Dropdown(label="Config/Split", choices=[])
view_data=gr.Button(label="View")
with gr.Row():
find_string=gr.Textbox()
find_btn=gr.Button("Search")
with gr.Row():
with gr.Column(scale=1):
out_json = gr.JSON()
with gr.Column(scale=3):
out_find = gr.JSON()
out_gal = gr.Gallery()
view_data.click(query2,[data_set_url,config_set,split_set],[out_find])
find_btn.click(find_fn,[find_string,data_set_url,config_set,split_set],[out_find])
fetch_btn.click(query1,data_set_url,out_json)
app.launch() |