remove ipython.display
Browse files- aimodel.py +1 -1
- test_rect.py +1 -1
aimodel.py
CHANGED
|
@@ -7,7 +7,6 @@ import torch
|
|
| 7 |
from pathlib import Path
|
| 8 |
from PIL import Image
|
| 9 |
from PIL import ImageDraw
|
| 10 |
-
from IPython.display import display
|
| 11 |
import numpy as np
|
| 12 |
from collections import namedtuple
|
| 13 |
from logging import getLogger
|
|
@@ -371,6 +370,7 @@ def load_cache():
|
|
| 371 |
#%%
|
| 372 |
if __name__ == '__main__':
|
| 373 |
from io import BytesIO
|
|
|
|
| 374 |
fl, fl_ft = model_init(hack=False)
|
| 375 |
#load_cache()
|
| 376 |
clear_cache()
|
|
|
|
| 7 |
from pathlib import Path
|
| 8 |
from PIL import Image
|
| 9 |
from PIL import ImageDraw
|
|
|
|
| 10 |
import numpy as np
|
| 11 |
from collections import namedtuple
|
| 12 |
from logging import getLogger
|
|
|
|
| 370 |
#%%
|
| 371 |
if __name__ == '__main__':
|
| 372 |
from io import BytesIO
|
| 373 |
+
from IPython.display import display
|
| 374 |
fl, fl_ft = model_init(hack=False)
|
| 375 |
#load_cache()
|
| 376 |
clear_cache()
|
test_rect.py
CHANGED
|
@@ -6,7 +6,6 @@ import torch
|
|
| 6 |
from pathlib import Path
|
| 7 |
from PIL import Image
|
| 8 |
from PIL import ImageDraw
|
| 9 |
-
from IPython.display import display
|
| 10 |
import numpy as np
|
| 11 |
from collections import namedtuple
|
| 12 |
import sys
|
|
@@ -134,6 +133,7 @@ def main():
|
|
| 134 |
print(red_i)
|
| 135 |
display(img2)
|
| 136 |
if __name__ == '__main__':
|
|
|
|
| 137 |
main()
|
| 138 |
|
| 139 |
#%%
|
|
|
|
| 6 |
from pathlib import Path
|
| 7 |
from PIL import Image
|
| 8 |
from PIL import ImageDraw
|
|
|
|
| 9 |
import numpy as np
|
| 10 |
from collections import namedtuple
|
| 11 |
import sys
|
|
|
|
| 133 |
print(red_i)
|
| 134 |
display(img2)
|
| 135 |
if __name__ == '__main__':
|
| 136 |
+
from IPython.display import display
|
| 137 |
main()
|
| 138 |
|
| 139 |
#%%
|