* Refactoring imports for tensorflow * Removing whitespace spaces on blank line 110 * Moving tensorflow to try/except block * Fixed black formatting * Refactoring try/except to if/else