Update app.py

app.py

@@ -5,6 +5,16 @@ import requests
 import json
 import spacy
 import spacy.cli
+import warnings
+import logging
+
+# Suppress torch warnings
+warnings.filterwarnings("ignore", message=".*torch.classes.*")
+warnings.filterwarnings("ignore", message=".*__path__._path.*")
+
+# Set logging level to reduce noise
+logging.getLogger("torch").setLevel(logging.ERROR)
+logging.getLogger("transformers").setLevel(logging.ERROR)
 
 # Page config
 st.set_page_config(
@@ -62,15 +72,19 @@ else:
 
 @st.cache_resource  # 👈 Add the caching decorator
 def load_model(selected_language, model_name=None, entity_set=None):
-    try:
-        if selected_language == "German":
-            # Download and load the German-specific model
-            try:
-                nlp_model_de = spacy.load("de_core_news_lg")
-            except OSError:
-                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")
+    # Suppress warnings during model loading
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore")
+        
+        try:
+            if selected_language == "German":
+                # Download and load the German-specific model
+                try:
+                    nlp_model_de = spacy.load("de_core_news_lg")
+                except OSError:
+                    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:
@@ -104,8 +118,10 @@ def load_model(selected_language, model_name=None, entity_set=None):
             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
+            with warnings.catch_warnings():
+                warnings.simplefilter("ignore")
+                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
@@ -258,5 +274,4 @@ if submit_button and entities_map:
     with st.expander("Here is the final JSON-LD"):
         st.json(json_ld_data)  # Output JSON-LD
 elif submit_button and not entities_map:
-    st.warning("No entities found in the text. Please try with different text or check if the model is working correctly.")
-    
+    st.warning("No entities found in the text. Please try with different text or check if the model is working correctly.")