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

Update app.py

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