Update classification.py
Browse files- classification.py +1 -1
classification.py
CHANGED
@@ -10,7 +10,7 @@ if gpus:
|
|
10 |
tf.config.experimental.set_memory_growth(gpu, True)
|
11 |
except RuntimeError as e:
|
12 |
# Memory growth must be set before GPUs have been initialized
|
13 |
-
print(e
|
14 |
import streamlit as st
|
15 |
# with tf.device('/cpu:0'):
|
16 |
# Load the saved model
|
|
|
10 |
tf.config.experimental.set_memory_growth(gpu, True)
|
11 |
except RuntimeError as e:
|
12 |
# Memory growth must be set before GPUs have been initialized
|
13 |
+
print(e)
|
14 |
import streamlit as st
|
15 |
# with tf.device('/cpu:0'):
|
16 |
# Load the saved model
|