github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-20 05:23:22 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

correctly identify when not in colab

Browse source
This commit is contained in:
Cameron Grant 2021-11-05 11:51:53 -07:00
commit 8b793c1779
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
core/interact/interact.py
View file

@ -12,14 +12,24 @@ from tqdm import tqdm
from core import stdex from core import stdex
is_colab = False
try: try:
import IPython #if success we are in colab import google.colab # if success we are in colab
is_colab = True
except ModuleNotFoundError:
pass
try:
from IPython import get_ipython
shell = get_ipython().__class__.__name__
if shell == 'ZMQInteractiveShell': # if true we are in a jupyter notebook
is_colab = True
except ModuleNotFoundError:
pass
if is_colab:
from IPython.display import display, clear_output from IPython.display import display, clear_output
import PIL import PIL
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
is_colab = True
except:
is_colab = False
yn_str = {True:'y',False:'n'} yn_str = {True:'y',False:'n'}