Update app.py

app.py

@@ -12,7 +12,12 @@ import textract
 def load_summarization_pipeline():
     try:
         # Initialize the summarization pipeline using the specified model.
-        summarizer = pipeline("summarization", model="deepseek-ai/deepseek-vl2-tiny")
+        # Adding trust_remote_code=True allows loading models with custom code.
+        summarizer = pipeline(
+            "summarization",
+            model="deepseek-ai/deepseek-vl2-tiny",
+            trust_remote_code=True
+        )
         return summarizer
     except Exception as e:
         st.error(f"Error loading summarization model: {e}")
@@ -79,7 +84,7 @@ def summarize_text(text):
         return "No text available to summarize."
     
     try:
-        # Note: The summarization pipeline might have limitations on text length.
+        # The summarization pipeline might have limitations on text length.
         # For long documents, consider splitting the text into smaller chunks.
         summary = summarizer(text, max_length=150, min_length=40, do_sample=False)
         return summary[0]["summary_text"]