CCCDev commited on
Commit
8882f97
·
verified ·
1 Parent(s): d698f36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- pipe = HuggingFacePipeline(pipeline("summarization", model=model, tokenizer=tokenizer))
 
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(