yuvaranianandhan24 commited on
Commit
4493731
·
verified ·
1 Parent(s): f314072

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "llama-2-7b-chat.ggmlv3.q4_0.bin"
42
  llm = HuggingFaceHub(
43
  huggingfacehub_api_token = api_token,
44
  repo_id = model,
45
- repo_type = "llama",
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