Update app.py

app.py

@@ -130,7 +130,8 @@ def load_saved_artifacts():
     global __model
     if __model is None:
         __model = joblib.load('saved_model.pkl') 
-    st.text("loading saved artifacts...done")
+    #st.text("loading saved artifacts...done")
+    return True
 
 def class_number_to_name(class_num):
     return __class_number_to_name[class_num]
@@ -157,36 +158,33 @@ if uploaded_image is not None:
     if image_dict["complete"]:
         display_image = image_dict["filename"]
         st.header("Image Uploded!, Processing...")
-        load_saved_artifacts()
-        
-        img = cv2.imread(display_image)
-        
-        
-       
-        result = classify_image(get_b64_test_image_for_virat(), display_image)
-        #st.text(result[0])
-        
-        col1,col2 = st.columns(2)
-        with col1:
-            st.image(img,width=150, caption='Uploaded Image')
-        with col2:
-            celeb = result[0]['class']
-            st.text("Your Predicted Sports Person is: " + result[0]['class'])
-            if celeb == "lionel_messi":
-                messi = cv2.imread("messi.jpeg")
-                st.image(messi,width=150, caption='Lionel Messi')
-            elif celeb == "maria_sharapova":
-                maria = cv2.imread("sharapova.jpeg")
-                st.image(maria,width=150, caption='Maria Sharapova')
-            elif celeb == "roger_federer":
-                roger = cv2.imread("federer.jpeg")
-                st.image(roger,width=150, caption='Roger Federer')
-            elif celeb == "serena_williams":
-                serena = cv2.imread("serena.jpeg")
-                st.image(serena,width=150, caption='Serena Williams')
-            elif celeb == "virat_kohli":
-                virat = cv2.imread("virat.jpeg")
-                st.image(virat,width=150, caption='Virat Kohli')
+        if load_saved_artifacts():
+            img = cv2.imread(display_image)
+            
+            result = classify_image(get_b64_test_image_for_virat(), display_image)
+            #st.text(result[0])
+            
+            col6,col7 = st.columns(2)
+            with col6:
+                st.image(img,width=130, caption='Uploaded Image')
+            with col7:
+                celeb = result[0]['class']
+                st.text("Your Predicted Sports Person is: " + result[0]['class'])
+                if celeb == "lionel_messi":
+                    messi = cv2.imread("messi.jpeg")
+                    st.image(messi,width=150, caption='Lionel Messi')
+                elif celeb == "maria_sharapova":
+                    maria = cv2.imread("sharapova.jpeg")
+                    st.image(maria,width=150, caption='Maria Sharapova')
+                elif celeb == "roger_federer":
+                    roger = cv2.imread("federer.jpeg")
+                    st.image(roger,width=150, caption='Roger Federer')
+                elif celeb == "serena_williams":
+                    serena = cv2.imread("serena.jpeg")
+                    st.image(serena,width=150, caption='Serena Williams')
+                elif celeb == "virat_kohli":
+                    virat = cv2.imread("virat.jpeg")
+                    st.image(virat,width=150, caption='Virat Kohli')