Fix progress bar compatibility with generators (#9693)

* Fix progress bar compatibility with generators * add changeset * Format helpers.py * Fix --------- Co-authored-by: Abubakar Abid <abubakar@huggingface.co> Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com> Co-authored-by: Freddy Boulton <alfonsoboulton@gmail.com>
gradio-app · Oct 15, 2024 · c45b466 · c45b466
1 parent 31418ef
commit c45b466
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/.changeset/khaki-sites-bow.md b/.changeset/khaki-sites-bow.md
@@ -0,0 +1,5 @@
+---
+"gradio": patch
+---
+
+fix:Fix progress bar compatibility with generators
diff --git a/gradio/helpers.py b/gradio/helpers.py
@@ -764,7 +764,7 @@ def tqdm(
                 self.iterables.append(new_iterable)
                 callback(self.iterables)
                 return self
-            length = len(iterable) if hasattr(iterable, "__len__") else None  # type: ignore
+            length = len(iterable) if hasattr(iterable, "__len__") else total  # type: ignore
             self.iterables.append(
                 TrackedIterable(iter(iterable), 0, length, desc, unit, _tqdm)
             )

diff --git a/gradio/queueing.py b/gradio/queueing.py
@@ -646,7 +646,7 @@ async def process_events(
                         event,
                         ProcessCompletedMessage(
                             output=content,
-                            title=content["title"],  # type: ignore
+                            title=content.get("title", "Error"),  # type: ignore
                             success=False,
                         ),
                     )