Spaces:

mery22
/

testing

Sleeping

mery22 commited on May 21, 2024

Commit

fe7c8df

verified ·

1 Parent(s): cb5c2b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ bnb_config = BitsAndBytesConfig(
 model = AutoModelForCausalLM.from_pretrained(
    "mistralai/Mistral-7B-Instruct-v0.1",quantization_config=bnb_config,
 )
-dataset= load_dataset("mery22/testub")
 loader = PyPDFLoader(dataset)
 data = loader.load()
 text_splitter1 = CharacterTextSplitter(chunk_size=512, chunk_overlap=0,separator="\n\n")

 model = AutoModelForCausalLM.from_pretrained(
    "mistralai/Mistral-7B-Instruct-v0.1",quantization_config=bnb_config,
 )
+dataset= load_dataset("mery22/testub/test-1.pdf")
 loader = PyPDFLoader(dataset)
 data = loader.load()
 text_splitter1 = CharacterTextSplitter(chunk_size=512, chunk_overlap=0,separator="\n\n")