Update app.py

app.py

@@ -41,12 +41,14 @@ def find_fn(inp,out_json):
     #print(out_json['rows'])
     #print (inp)
     img_list=[]
-    out_json1=query2("",offset=10,length=20)
-    for rows in out_json1['rows']:
-        img_ea = rows['row']['image']['src']
-        img_list.append(img_ea)
+    #out_json1=query2(,offset=10,length=20)
     
-    return img_list,img_list
+    inp=inp.strip("[]")
+    for ea in out_json[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)")