Update app.py

app.py

@@ -45,9 +45,9 @@ def find_fn(inp,fetch_url,config="TEST",split="test",offset=0,length=10):
     
     inp=inp.strip("[]")
     print(inp)
-    for ea in out_json[f"{inp}"]:
+    for ea in out_json:
         #img_ea = ea['row']['image']['src']
-        img_list.append(ea)
+        img_list.append(ea[f"{inp}"])
     
     return img_list
 with gr.Blocks() as app: