Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -66,8 +66,9 @@ 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 |
if query == 'Summarize':
|
| 72 |
#docs = pdf_reader.load_and_split()
|
| 73 |
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 |
+
#,
|
| 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")
|