Update src/streamlit_app.py

src/streamlit_app.py

@@ -31,7 +31,7 @@ class RepVGGBlock(layers.Layer):
         self.actual_in_channels = None
 
         self.rbr_dense_conv = layers.Conv2D(
-            filters=self.config_out_out_channels, kernel_size=self.config_kernel_size,
+            filters=self.config_out_channels, kernel_size=self.config_kernel_size,
             strides=self.config_strides_val, padding='same',
             groups=self.config_groups, use_bias=False, name=self.name + '_dense_conv'
         )
@@ -240,15 +240,16 @@ class NECALayer(layers.Layer):
 # --- End of NECALayer ---
 
 # --- Streamlit App Configuration ---
-MODEL_FILENAME = 'genera_cic_v1.keras'
-LABEL_MAPPING_FILENAME = 'label_mapping.json'
+# Updated paths to reflect the file location within the Hugging Face Space 'src' directory
+MODEL_FILENAME = '/app/src/genera_cic_v1.keras'
+LABEL_MAPPING_FILENAME = '/app/src/label_mapping.json'
 IMG_WIDTH = 299
 IMG_HEIGHT = 299
 
 st.set_page_config(page_title="Genera Cloud Classifier", layout="wide")
 
 # --- Load Model and Label Mapping (Cached for performance) ---
-@st.cache_resource # Added newline here
+@st.cache_resource
 def load_keras_model(model_path):
     """Loads the Keras model with custom layer definitions."""
     if not os.path.exists(model_path):