Update app.py
Browse files
app.py
CHANGED
@@ -38,11 +38,11 @@ def process_pdf(uploaded_file):
|
|
38 |
for page in pages:
|
39 |
st.write(page.page_content)
|
40 |
|
41 |
-
model = "
|
42 |
llm = HuggingFaceHub(
|
43 |
huggingfacehub_api_token = api_token,
|
44 |
repo_id = model,
|
45 |
-
repo_type = "
|
46 |
verbose = False,
|
47 |
model_kwargs = {"temperature":0.01, "max_new_tokens": 128})
|
48 |
|
|
|
38 |
for page in pages:
|
39 |
st.write(page.page_content)
|
40 |
|
41 |
+
model = "TheBloke/Llama-2-7B-Chat-GGML"
|
42 |
llm = HuggingFaceHub(
|
43 |
huggingfacehub_api_token = api_token,
|
44 |
repo_id = model,
|
45 |
+
repo_type = "text-generation",
|
46 |
verbose = False,
|
47 |
model_kwargs = {"temperature":0.01, "max_new_tokens": 128})
|
48 |
|