Spaces:

inesctec
/

Boundary-Detection-Demo

Running

jmisidro commited on 20 days ago

Commit

51a33b0

verified ·

1 Parent(s): 89ef970

Update src/streamlit_app.py

Files changed (1) hide show

src/streamlit_app.py CHANGED Viewed

@@ -96,8 +96,8 @@ st.markdown("""
 # ==================== MODEL LOADING ====================
 MODEL_OPTIONS = {
-    "Structural Segmentation Baseline": "liaad/Citilink-XLMR-large-Structural-Segmentation-pt",
-    "Structural Segmentation Augmented": "liaad/Citilink-XLMR-large-Structural-Segmentation-pt-augmented",
 }
 @st.cache_resource

 # ==================== MODEL LOADING ====================
 MODEL_OPTIONS = {
+    "Structural Segmentation Baseline": "inesctec/Citilink-XLMR-large-Structural-Segmentation-pt",
+    "Structural Segmentation Augmented": "inesctec/Citilink-XLMR-large-Structural-Segmentation-pt-augmented",
 }
 @st.cache_resource