Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ def process_vid(file,cur_frame,every_n):
|
|
37 |
frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
|
38 |
rev_img_searcher = ReverseImageSearcher()
|
39 |
html_out=""
|
40 |
-
count =
|
41 |
if cur_frame == "" or cur_frame==None:
|
42 |
start_frame = 0
|
43 |
elif cur_frame != "" and cur_frame!=None:
|
@@ -73,14 +73,15 @@ def process_vid(file,cur_frame,every_n):
|
|
73 |
Img: <a href='{search_item.image_url}' target='_blank' rel='noopener noreferrer'>{search_item.image_url}</a><br>
|
74 |
<img class='my_im' src='{search_item.image_url}'><br>
|
75 |
</div>"""
|
76 |
-
|
|
|
77 |
else:
|
78 |
pass
|
79 |
count +=1
|
80 |
yield (None,f"Searching Frame: {i}", "")
|
81 |
except Exception as e:
|
82 |
-
|
83 |
-
|
84 |
|
85 |
def process_im(file):
|
86 |
read_file = Image.open(file)
|
|
|
37 |
frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
|
38 |
rev_img_searcher = ReverseImageSearcher()
|
39 |
html_out=""
|
40 |
+
count = int(every_n)
|
41 |
if cur_frame == "" or cur_frame==None:
|
42 |
start_frame = 0
|
43 |
elif cur_frame != "" and cur_frame!=None:
|
|
|
73 |
Img: <a href='{search_item.image_url}' target='_blank' rel='noopener noreferrer'>{search_item.image_url}</a><br>
|
74 |
<img class='my_im' src='{search_item.image_url}'><br>
|
75 |
</div>"""
|
76 |
+
i=frame_count-1
|
77 |
+
yield (gr.HTML(f'<h1>Total Found: {count}</h1><br>{html_out}'), f"Found frame: {i}", i)
|
78 |
else:
|
79 |
pass
|
80 |
count +=1
|
81 |
yield (None,f"Searching Frame: {i}", "")
|
82 |
except Exception as e:
|
83 |
+
yield (gr.HTML(f'{e}'),"","")
|
84 |
+
yield (gr.HTML('No frame matches found.'),"","")
|
85 |
|
86 |
def process_im(file):
|
87 |
read_file = Image.open(file)
|