Update app.py

app.py

@@ -32,9 +32,13 @@ try:
     logger.info("Successfully loaded RagRetriever with trust_remote_code=True")
 except ValueError as e:
     logger.error(f"Error loading RagRetriever: {e}")
-    st.error(f"Error loading RagRetriever: {e}")
+    rag_retriever = None
 
-rag_model = RagSequenceForGeneration.from_pretrained(rag_model_name, retriever=rag_retriever)
+if rag_retriever is not None:
+    rag_model = RagSequenceForGeneration.from_pretrained(rag_model_name, retriever=rag_retriever)
+else:
+    logger.error("RagRetriever is not available, unable to proceed with loading RAG model.")
+    st.error("RagRetriever is not available, unable to proceed with loading RAG model.")
 
 def transcribe_audio(audio_path, language="ru"):
     speech, rate = librosa.load(audio_path, sr=16000)
@@ -45,6 +49,8 @@ def transcribe_audio(audio_path, language="ru"):
     return transcription
 
 def translate_and_summarize(text):
+    if rag_retriever is None:
+        return ["Translation and summarization feature is not available due to RAG retriever loading issue."]
     inputs = rag_tokenizer(text, return_tensors="pt")
     input_ids = inputs["input_ids"]
     attention_mask = inputs["attention_mask"]