Update app.py

app.py

@@ -25,26 +25,21 @@ load_dotenv()
 shared_db = lancedb.connect("~/rag_demo_db")
 
 
-@spaces.GPU
-def get_all_model():
-    # Initialize embedding models
-    # text_embedding_model = SentenceTransformer("all-MiniLM-L6-v2", trust_remote_code=True)
-    text_embedding_model = SentenceTransformer(
-        "BAAI/bge-base-en", trust_remote_code=True
-    )
-    # text_embedding_model = SentenceTransformer("BAAI/bge-large-en-v1.5", trust_remote_code=True)
-    # text_embedding_model = SentenceTransformer("BAAI/bge-small-en-v1.5", trust_remote_code=True)
-    image_embedding_model = SentenceTransformer(
-        "jinaai/jina-clip-v1", trust_remote_code=True
-    )
-    colpali_model, colpali_processor = get_model_colpali("vidore/colpali-v1.2")
-
-    return text_embedding_model, image_embedding_model, colpali_model, colpali_processor
 
 
-text_embedding_model, image_embedding_model, colpali_model, colpali_processor = (
-    get_all_model()
+# Initialize embedding models
+# text_embedding_model = SentenceTransformer("all-MiniLM-L6-v2", trust_remote_code=True)
+text_embedding_model = SentenceTransformer(
+    "BAAI/bge-base-en", trust_remote_code=True
 )
+# text_embedding_model = SentenceTransformer("BAAI/bge-large-en-v1.5", trust_remote_code=True)
+# text_embedding_model = SentenceTransformer("BAAI/bge-small-en-v1.5", trust_remote_code=True)
+image_embedding_model = SentenceTransformer(
+    "jinaai/jina-clip-v1", trust_remote_code=True
+)
+colpali_model, colpali_processor = get_model_colpali("vidore/colpali-v1.2")
+
+
 
 # Initialize RAG instances
 simple_rag = SimpleRAG(