nirmalaag commited on
Commit
07b7ab2
·
verified ·
1 Parent(s): 78d3202

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,8 +64,8 @@ def main():
64
  if query:
65
 
66
  #PATH = 'model/'
67
- llm = AutoModelForCausalLM.from_pretrained("openai-community/gpt2-medium")
68
- #llm = huggingface_hub.HuggingFaceHub(repo_id="openai-community/gpt2",model_kwargs={"temperature":1.0, "max_length":500})
69
  if query == 'Summarize':
70
  #docs = pdf_reader.load_and_split()
71
  chain = load_summarize_chain(llm, chain_type="map_reduce")
 
64
  if query:
65
 
66
  #PATH = 'model/'
67
+ #llm = AutoModelForCausalLM.from_pretrained("openai-community/gpt2-medium")
68
+ llm = huggingface_hub.HuggingFaceHub(repo_id="openai-community/gpt2-medium",model_kwargs={"temperature":1.0, "max_length":500})
69
  if query == 'Summarize':
70
  #docs = pdf_reader.load_and_split()
71
  chain = load_summarize_chain(llm, chain_type="map_reduce")