Update app.py

app.py

@@ -95,13 +95,16 @@ def process_im(file,url):
         out_url = f'https://omnibus-reverse-image.hf.space/file={out}'
         return (out_url)
 
-def rev_im(image_url):
+def rev_im(image):
     #image_url = 'https://i.pinimg.com/originals/c4/50/35/c450352ac6ea8645ead206721673e8fb.png'
     out_list = []
     out_im = []
     html_out = """"""
+    cv2.imwrite(f"{uid}-im_tmp.png", image)
+    out = os.path.abspath(f"{uid}-im_tmp.png")
+    out_url = f'https://omnibus-reverse-image-dev.hf.space/file={out}'    
     rev_img_searcher = ReverseImageSearcher()
-    res = rev_img_searcher.search(image_url)
+    res = rev_img_searcher.search(out_url)
     count = 0
     for search_item in res:
         count+=1
@@ -164,8 +167,8 @@ with gr.Blocks() as app:
     next_btn.click(process_vid,[inp_vid,hid_box,every_n],[html_out,stat_box,hid_box])
     vid_url_btn.click(dl,vid_url,[inp_vid,html_out,stat_box,hid_box])
     source_tog.change(shuf,[source_tog],[im_box,vid_box])
-    inp_im.change(process_im,[inp_im,inp_url],[inp_url])
+    #inp_im.change(process_im,[inp_im,inp_url],[inp_url])
     go_btn_vid.click(process_vid,[inp_vid,hid_box,every_n],[html_out,stat_box,hid_box])
-    go_btn_im.click(rev_im,inp_url,[html_out])
+    go_btn_im.click(rev_im,inp_im,[html_out])
     #go_btn_url.click(rev_im,inp_url,[html_out])
 app.launch()