Update simple_app.py
Browse files- simple_app.py +5 -4
simple_app.py
CHANGED
|
@@ -37,17 +37,18 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
|
|
| 37 |
progress_pattern = re.compile(r"(\d+)%\|.*\| (\d+)/(\d+)")
|
| 38 |
|
| 39 |
for line in iter(process.stdout.readline, ''):
|
| 40 |
-
# Try to parse progress info from the line
|
| 41 |
match = progress_pattern.search(line)
|
| 42 |
if match:
|
|
|
|
| 43 |
current = int(match.group(2))
|
| 44 |
total = int(match.group(3))
|
| 45 |
if progress_bar is None:
|
| 46 |
-
progress_bar = tqdm(total=total, desc="Video Generation Progress")
|
| 47 |
-
# Update the progress bar only if progress has advanced
|
| 48 |
progress_bar.update(current - progress_bar.n)
|
|
|
|
|
|
|
| 49 |
else:
|
| 50 |
-
#
|
| 51 |
print(line, end="")
|
| 52 |
|
| 53 |
process.wait()
|
|
|
|
| 37 |
progress_pattern = re.compile(r"(\d+)%\|.*\| (\d+)/(\d+)")
|
| 38 |
|
| 39 |
for line in iter(process.stdout.readline, ''):
|
|
|
|
| 40 |
match = progress_pattern.search(line)
|
| 41 |
if match:
|
| 42 |
+
# Update tqdm without printing the line to logs
|
| 43 |
current = int(match.group(2))
|
| 44 |
total = int(match.group(3))
|
| 45 |
if progress_bar is None:
|
| 46 |
+
progress_bar = tqdm(total=total, desc="Video Generation Progress", leave=True)
|
|
|
|
| 47 |
progress_bar.update(current - progress_bar.n)
|
| 48 |
+
# Optionally, refresh the tqdm display without a newline:
|
| 49 |
+
progress_bar.refresh()
|
| 50 |
else:
|
| 51 |
+
# For non-progress lines, print them normally
|
| 52 |
print(line, end="")
|
| 53 |
|
| 54 |
process.wait()
|