Spaces:

EmanHassan26
/

RAGMultiModalEx

Sleeping

EmanHassan26 commited on 24 days ago

Commit

82cd3cf

verified ·

1 Parent(s): f904bfe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def load_or_init_rag():
         return RAGMultiModalModel.from_index(INDEX_NAME)
     else:
         print("No index found. Initializing vanilla model context wrapper.")
-        return RAGMultiModalModel.from_pretrained("vidore/colqwen2-v1.0")
 RAG = load_or_init_rag()

         return RAGMultiModalModel.from_index(INDEX_NAME)
     else:
         print("No index found. Initializing vanilla model context wrapper.")
+        return RAGMultiModalModel.from_pretrained("vidore/colqwen2-v1.0",device=device)
 RAG = load_or_init_rag()