Update app.py

app.py

@@ -4,8 +4,8 @@ import os
 import random
 import time
 
-os.system("pip install gradio, llama_index, ragatouille, llama-cpp-python")
-os.system("git clone https://github.com/EnPaiva93/think-paraguayo-space-aux.git")
+# os.system("pip install gradio, llama_index, ragatouille, llama-cpp-python")
+# os.system("git clone https://github.com/EnPaiva93/think-paraguayo-space-aux.git")
 os.system("wget https://huggingface.co/thinkPy/gua-a_v0.2-dpo_mistral-7b_GGUF/resolve/main/gua-a_v0.2-dpo_mistral-7b_q4_K_M.gguf -O model.gguf")
 
 from llama_cpp import Llama
@@ -46,16 +46,11 @@ llm = Llama(model_path="model.gguf",
             n_ctx=512,
             n_threads=2)
 
-BASE_PATH = "/home/user/app/think-paraguayo-space-aux/"
-
-DOC_PATH = BASE_PATH+"index/" 
+DOC_PATH = ".ragatouille/colbert/indexes/index" 
 
 print(os.listdir())
 
-os.system("sudo mkdir /content/")
-os.system("sudo cp -r ./think-paraguayo-space-aux/ /content/")
-
-documents = SimpleDirectoryReader(input_files=[BASE_PATH+"libro.txt"]).load_data()
+documents = SimpleDirectoryReader(input_files=["libro.txt"]).load_data()
 
 parser = SentenceSplitter(chunk_size=128, chunk_overlap=64)
 nodes = parser.get_nodes_from_documents(
@@ -68,6 +63,7 @@ print(os.getcwd())
 # if os.path.exists(DOC_PATH):
 RAG = RAGPretrainedModel.from_pretrained("AdrienB134/ColBERTv2.0-spanish-mmarcoES")
 RAG = RAG.from_index(DOC_PATH, n_gpu=None)
+RAG.search("init", None, k=1)
 
 # else:
 #   RAG = RAGPretrainedModel.from_pretrained("AdrienB134/ColBERTv2.0-spanish-mmarcoES")
@@ -153,7 +149,7 @@ def launcher():
 
         with gr.Row(variant='panel'):
             with gr.Column(scale=1):
-                gr.Image(value=BASE_PATH+"think_paraguayo.jpeg", type="filepath", label="Imagen Estática")
+                gr.Image(value="think_paraguayo.jpeg", type="filepath", label="Imagen Estática")
 
             with gr.Column(scale=1):
                 # with gr.Row():