File size: 1,499 Bytes
d714d3d
f894099
60ed47f
49687e4
 
ed88b70
 
49687e4
8d217ca
1d06fdf
f894099
8d217ca
 
f894099
d714d3d
 
f894099
 
d714d3d
 
 
 
 
 
f894099
60ed47f
4ec3536
f894099
ef02a4a
f597636
ef02a4a
a8a0883
3840b5a
a8a0883
 
 
 
 
ef02a4a
 
 
 
 
 
 
 
a8a0883
 
 
ef02a4a
a8a0883
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
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()