OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-28 06:59:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Revert "added rich console patch for Google Colab"

Browse Source 
This reverts commit db0b87597d.
This commit is contained in:
k4yt3x 2022-09-04 21:56:40 +00:00
parent db0b87597d
commit 84b730497b
1 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
video2x/video2x.py
View File

@ -27,7 +27,7 @@ __ __ _ _ ___ __ __
Name: Video2X
Creator: K4YT3X
Date Created: February 24, 2018
Last Modified: September 4, 2022
Last Modified: August 28, 2022
Editor: BrianPetkovsek
Last Modified: June 17, 2019
@ -41,7 +41,6 @@ Last Modified: March 23, 2020
import ctypes
import math
import os
import signal
import sys
import time
@ -165,14 +164,8 @@ class Video2X:
original_stdout = sys.stdout
original_stderr = sys.stderr
# detect if the program is running in Google Colab
# COLAB_GPU should be set in Colab runtime environments
in_colab = os.environ.get("COLAB_GPU") is not None
# create console for rich's Live display
# force_jupyter must be True if the script is running in Colab
# other wise the console will not print properly
console = Console(force_jupyter=True if in_colab is True else None)
console = Console()
# redirect STDOUT and STDERR to console
sys.stdout = FileProxy(console, sys.stdout)
@ -440,4 +433,4 @@ class Video2X:
ProcessingMode.INTERPOLATE,
processes,
(threshold, algorithm),
)
)