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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -42,6 +42,7 @@ def process_pdf(uploaded_file):
42
  llm = HuggingFaceHub(
43
  huggingfacehub_api_token = api_token,
44
  repo_id = model,
 
45
  verbose = False,
46
  model_kwargs = {"temperature":0.01, "max_new_tokens": 128})
47
 
@@ -76,6 +77,7 @@ def process_image(uploaded_image):
76
  llm = HuggingFaceHub(
77
  huggingfacehub_api_token = api_token,
78
  repo_id = model,
 
79
  verbose = False,
80
  model_kwargs = {"temperature":0.01, "max_new_tokens": 128})
81
 
 
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
 
 
77
  llm = HuggingFaceHub(
78
  huggingfacehub_api_token = api_token,
79
  repo_id = model,
80
+ repo_type = 'llama',
81
  verbose = False,
82
  model_kwargs = {"temperature":0.01, "max_new_tokens": 128})
83