Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
@@ -77,37 +77,7 @@ class InferenceRunner(StateManager):
|
|
77 |
st.session_state['settings_changed'] = self.has_state_changed()
|
78 |
if st.session_state['settings_changed']:
|
79 |
self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
###################################
|
87 |
-
self.col1.write("Choose from sample images:")
|
88 |
-
cols = self.col1.columns(len(self.sample_images))
|
89 |
-
for idx, sample_image_path in enumerate(self.sample_images):
|
90 |
-
with cols[idx]:
|
91 |
-
image = Image.open(sample_image_path)
|
92 |
-
image_for_display = copy.deepcopy(image) # resize just for the display control without changing original image
|
93 |
-
image_for_display = self.resize_image(image_for_display, 500, 500)
|
94 |
-
st.image(image_for_display)
|
95 |
-
|
96 |
-
|
97 |
-
# Image uploader
|
98 |
-
uploaded_image = self.col1.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
|
99 |
-
image_for_display = copy.deepcopy(uploaded_image) # resize just for the display control without changing original image
|
100 |
-
image_for_display = self.resize_image(image_for_display, 500, 500)
|
101 |
-
|
102 |
-
############################################
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
|
112 |
st.session_state.button_label = "Reload Model" if self.is_model_loaded() and self.settings_changed else "Load Model"
|
113 |
|
|
|
77 |
st.session_state['settings_changed'] = self.has_state_changed()
|
78 |
if st.session_state['settings_changed']:
|
79 |
self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
|
80 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
81 |
|
82 |
st.session_state.button_label = "Reload Model" if self.is_model_loaded() and self.settings_changed else "Load Model"
|
83 |
|