KalooAI commited on
Commit
d9ab9ba
·
verified ·
1 Parent(s): b8cf42f

Update app.py

Browse files

corrected 'pipeline' to 'pipe' in get_llm()

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -62,7 +62,7 @@ def get_llm():
62
  torch_dtype='float32',
63
  )
64
 
65
- pipeline = pipeline(
66
  model=model,
67
  tokenizer=tokenizer,
68
  max_new_tokens=256,
@@ -70,7 +70,7 @@ def get_llm():
70
  do_sample= True
71
  )
72
 
73
- return HuggingFacePipeline(pipeline=pipeline)
74
 
75
  ## Document loader
76
  def document_loader(file):
 
62
  torch_dtype='float32',
63
  )
64
 
65
+ pipe = pipeline(
66
  model=model,
67
  tokenizer=tokenizer,
68
  max_new_tokens=256,
 
70
  do_sample= True
71
  )
72
 
73
+ return HuggingFacePipeline(pipeline=pipe)
74
 
75
  ## Document loader
76
  def document_loader(file):