Update app.py

app.py

@@ -37,7 +37,7 @@ def process_vid(file,cur_frame,every_n):
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
     rev_img_searcher = ReverseImageSearcher()
     html_out=""
-    count = 1
+    count = int(every_n)
     if cur_frame == "" or cur_frame==None:
         start_frame = 0
     elif cur_frame != "" and cur_frame!=None:
@@ -73,14 +73,15 @@ def process_vid(file,cur_frame,every_n):
                         Img: <a href='{search_item.image_url}' target='_blank' rel='noopener noreferrer'>{search_item.image_url}</a><br>
                         <img class='my_im' src='{search_item.image_url}'><br>
                         </div>"""
-                    return (gr.HTML(f'<h1>Total Found: {count}</h1><br>{html_out}'), f"Found frame: {i}", i)
+                    i=frame_count-1
+                    yield (gr.HTML(f'<h1>Total Found: {count}</h1><br>{html_out}'), f"Found frame: {i}", i)
                 else:
                     pass
             count +=1
             yield (None,f"Searching Frame: {i}", "")
     except Exception as e:
-        return (gr.HTML(f'{e}'),"","")
-    return (gr.HTML('No frame matches found.'),"","")
+        yield (gr.HTML(f'{e}'),"","")
+    yield (gr.HTML('No frame matches found.'),"","")
 
 def process_im(file):
     read_file = Image.open(file)