Update app.py

app.py

@@ -58,7 +58,7 @@ with gr.Blocks() as demo:
             with gr.Row():
                 model_id = gr.Radio(["Zephyr-7b-Beta", "Llama-2-7b-chat"], value="Llama-2-7b-chat",label="LLM Model")
             with gr.Row():    
-                mode = gr.Radio(['Document', 'Data'], value='Document',label="QA mode")
+                mode = gr.Radio(['OITF Manuals', 'Operations Data'], value='OITF Manuals',label="Data")
         persist_directory = "db"
         embeddings = HuggingFaceBgeEmbeddings(
             model_name = "BAAI/bge-small-en-v1.5",
@@ -102,7 +102,7 @@ with gr.Blocks() as demo:
                 }
             )
             history[-1][1] = res['answer']
-            torch.cuda.empty_cache()
+            # torch.cuda.empty_cache()
             return history
         with gr.Column(scale=9): # type: ignore
             with gr.Row():