nirmalaag commited on
Commit
6fa5a85
·
verified ·
1 Parent(s): 37138f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -66,9 +66,8 @@ def main():
66
  #PATH = 'model/'
67
  #llm = AutoModelForCausalLM.from_pretrained("CohereForAI/aya-101")
68
  # llm = AutoModelForCausalLM.from_pretrained(PATH,local_files_only=True)
69
- llm = huggingface_hub.HuggingFaceHub(repo_id="openai-community/gpt2")
70
- #,
71
- # model_kwargs={"temperature":1.0, "max_length":256})
72
  if query == 'Summarize':
73
  #docs = pdf_reader.load_and_split()
74
  chain = load_summarize_chain(llm, chain_type="map_reduce")
 
66
  #PATH = 'model/'
67
  #llm = AutoModelForCausalLM.from_pretrained("CohereForAI/aya-101")
68
  # llm = AutoModelForCausalLM.from_pretrained(PATH,local_files_only=True)
69
+ llm = huggingface_hub.HuggingFaceHub(repo_id="openai-community/gpt2"),
70
+ model_kwargs={"temperature":1.0, "max_length":1000})
 
71
  if query == 'Summarize':
72
  #docs = pdf_reader.load_and_split()
73
  chain = load_summarize_chain(llm, chain_type="map_reduce")