Update app.py

app.py

@@ -48,15 +48,10 @@ with st.sidebar:
     """)
 st.set_option("deprecation.showfileUploaderEncoding", False)
 
-
 @st.cache_resource(show_spinner=False)
 def load_model():
     try:
         from huggingface_hub import from_pretrained_keras
-        config = tf.compat.v1.ConfigProto()
-        config.gpu_options.allow_growth = True
-        tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))
-        
         keras.utils.set_random_seed(42)
         model = from_pretrained_keras("ryefoxlime/PneumoniaDetection")
         return model
@@ -64,7 +59,6 @@ def load_model():
         st.error(f"Error loading model: {str(e)}")
         return None
 
-
 with st.spinner("Model is being loaded..."):
     model = load_model()