Update app.py

app.py

@@ -5,19 +5,25 @@ from mhr.predict_tools import MHRStoneRecognizeMgr, MHRVedioCuter
 mgr = MHRStoneRecognizeMgr(
     whole_pkl = "./whole_model.pkl",
     hole_pkl = "./hole_model.pkl",
-    vedio_cutter = MHRVedioCuter(5),
+    vedio_cutter = None,
 )
 
 def greet(v):
-    print(v)
-    ret = mgr.recognize(v)
-    print(ret)
-    return ret
+    result = []
+    progress = gr.Progress(0)
+    cuter = MHRVedioCuter(5)
+    for data in progress.tqdm(cuter.iter(v)):
+        ok, ret = mgr.recognize_image(v)
+        print(ret)
+        result.append(ret)
+        
+    return result
 
 demo = gr.Interface(
     fn=greet,
     inputs=gr.Video(),
     outputs="text",
+    examples = "./example",
 )
 
-demo.launch()
+demo.launch(debug=True)