Update app.py

app.py

@@ -85,46 +85,45 @@ def load_model(selected_language, model_name=None, entity_set=None):
                     st.info("Downloading German language model... This may take a moment.")
                     spacy.cli.download("de_core_news_lg")
                     nlp_model_de = spacy.load("de_core_news_lg")
-            
-            # Check if entityfishing component is available
-            if "entityfishing" not in nlp_model_de.pipe_names:
-                try:
-                    nlp_model_de.add_pipe("entityfishing")
-                except Exception as e:
-                    st.warning(f"Entity-fishing not available, using basic NER only: {e}")
-                    # Return model without entityfishing for basic NER
-                    return nlp_model_de
-            
-            return nlp_model_de
-            
-        elif selected_language == "English - spaCy":
-            # Download and load English-specific model
-            try:
-                nlp_model_en = spacy.load("en_core_web_sm")
-            except OSError:
-                st.info("Downloading English language model... This may take a moment.")
-                spacy.cli.download("en_core_web_sm")
-                nlp_model_en = spacy.load("en_core_web_sm")
-            
-            # Check if entityfishing component is available
-            if "entityfishing" not in nlp_model_en.pipe_names:
+                
+                # Check if entityfishing component is available
+                if "entityfishing" not in nlp_model_de.pipe_names:
+                    try:
+                        nlp_model_de.add_pipe("entityfishing")
+                    except Exception as e:
+                        st.warning(f"Entity-fishing not available, using basic NER only: {e}")
+                        # Return model without entityfishing for basic NER
+                        return nlp_model_de
+                
+                return nlp_model_de
+                
+            elif selected_language == "English - spaCy":
+                # Download and load English-specific model
                 try:
-                    nlp_model_en.add_pipe("entityfishing")
-                except Exception as e:
-                    st.warning(f"Entity-fishing not available, using basic NER only: {e}")
-                    # Return model without entityfishing for basic NER
-                    return nlp_model_en
+                    nlp_model_en = spacy.load("en_core_web_sm")
+                except OSError:
+                    st.info("Downloading English language model... This may take a moment.")
+                    spacy.cli.download("en_core_web_sm")
+                    nlp_model_en = spacy.load("en_core_web_sm")
+                
+                # Check if entityfishing component is available
+                if "entityfishing" not in nlp_model_en.pipe_names:
+                    try:
+                        nlp_model_en.add_pipe("entityfishing")
+                    except Exception as e:
+                        st.warning(f"Entity-fishing not available, using basic NER only: {e}")
+                        # Return model without entityfishing for basic NER
+                        return nlp_model_en
 
-            return nlp_model_en    
-        else:
-            # Load the pretrained model for other languages
-            with warnings.catch_warnings():
-                warnings.simplefilter("ignore")
+                return nlp_model_en    
+            else:
+                # Load the pretrained model for other languages
                 refined_model = Refined.from_pretrained(model_name=model_name, entity_set=entity_set)
                 return refined_model
-    except Exception as e:
-        st.error(f"Error loading model: {e}")
-        return None
+                
+        except Exception as e:
+            st.error(f"Error loading model: {e}")
+            return None
 
 # Use the cached model
 model = load_model(selected_language, selected_model_name, selected_entity_set)