File size: 395 Bytes
048f18b
 
 
 
 
fa1f2cc
048f18b
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from langchain.llms import CTransformers
from langchain_experimental.agents.agent_toolkits.csv.base import create_csv_agent

llm = CTransformers(model="TheBloke/Llama-2-13B-Ensemble-v5-GGUF", model_file="llama-2-13b-ensemble-v5.Q5_K_M.gguf", model_type="llama")

csv_file = "data_breast.csv"

agent = create_csv_agent(llm, csv_file, verbose=True)

def quiz(query):
    return agent.run(query)