Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,8 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
|
|
49 |
|
50 |
tokenizer = AutoTokenizer.from_pretrained(llm_model)
|
51 |
model = AutoModelForSeq2SeqLM.from_pretrained(llm_model)
|
52 |
-
|
|
|
53 |
|
54 |
progress(0.75, desc="Defining buffer memory...")
|
55 |
memory = ConversationBufferMemory(
|
|
|
49 |
|
50 |
tokenizer = AutoTokenizer.from_pretrained(llm_model)
|
51 |
model = AutoModelForSeq2SeqLM.from_pretrained(llm_model)
|
52 |
+
summarization_pipeline = pipeline("summarization", model=model, tokenizer=tokenizer)
|
53 |
+
pipe = HuggingFacePipeline(pipeline=summarization_pipeline)
|
54 |
|
55 |
progress(0.75, desc="Defining buffer memory...")
|
56 |
memory = ConversationBufferMemory(
|