add progress counter

app.py

@@ -42,19 +42,24 @@ def flatten(l):
     return [item for sublist in l for item in sublist]
 
 
-def run_model_with_frames(text, framesize=4, overlap=3):
+def run_model_with_frames(text, framesize=4, overlap=3, progress=gr.Progress()):
     if overlap > framesize:
         return "Overlap should be smaller than batch size"
     frames = create_frames(text, framesize, overlap)
+    progress(0, desc="Starting...", total=len(frames))
+    counter = 0
     result = set()
     for frame in frames:
         questions = flatten(hf_run_model(frame))
         for question in questions:
             result.add(question.strip())
+        counter += 1
+        progress(counter, desc="Generating...")
     output_string = ""
     for entry in result:
         output_string += entry
         output_string += "\n"
+    progress=(counter, desc="Done")
     return output_string