Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -66,7 +66,7 @@ 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",model_kwargs={"temperature":1.0, "max_length":
|
70 |
if query == 'Summarize':
|
71 |
#docs = pdf_reader.load_and_split()
|
72 |
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",model_kwargs={"temperature":1.0, "max_length":500})
|
70 |
if query == 'Summarize':
|
71 |
#docs = pdf_reader.load_and_split()
|
72 |
chain = load_summarize_chain(llm, chain_type="map_reduce")
|