Update app.py

app.py

@@ -160,29 +160,8 @@ def main(query: str, client: QdrantClient, collection_name: str, tokenizer: Mist
 @st.cache_resource
 def load_models_and_documents():
     with st.spinner('Load models...'):
-        model_path = snapshot_download(
-            repo_id="Ichigo2899/mistralai-Mistral-Nemo-Instruct-2407-AWQ"
-        )
-        with tempfile.TemporaryDirectory() as tmpdirname:
-            snapshot_download(
-                repo_id='mistralai/Mistral-Nemo-Instruct-2407', 
-                allow_patterns=["params.json", 'tekken.json'], 
-                token=os.getenv('AccessMistralNemo'),
-                cache_dir=tmpdirname
-            )
-            for root, dirs, files in os.walk(tmpdirname):
-                for file in files:
-                    if file in ["params.json", "tekken.json"]:
-                        dest_path = os.path.join(model_path, file)
-                        if os.path.exists(dest_path):
-                            print(f"{dest_path} already exists, skipping or renaming")
-                        else:
-                            shutil.copy(file, model_path)
-
-        if os.path.exists(f"{model_path}/tekken.json"):
-            print('EXIST')
-        else:
-            print('DOESNT EXIST')
+        model_path = snapshot_download(repo_id="GameScribes/Mistral-Nemo-AWQ")
+        
         tokenizer = MistralTokenizer.from_file(f"{model_path}/tekken.json")
         
         llm = vllm.LLM(