KvrParaskevi commited on
Commit
8ab1cc5
·
verified ·
1 Parent(s): aca755b

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +3 -3
chatbot.py CHANGED
@@ -52,9 +52,9 @@ def load_pipeline():
52
  temperature = 0.1,repetition_penalty = 1.03)
53
 
54
  llm = HuggingFacePipeline.from_model_id(
55
- model_id=my_model_id,
56
  task="text-generation",
57
- model_kwargs={"trust_remote_code": True},
58
  pipeline_kwargs={
59
  "model": model,
60
  "tokenizer": tokenizer,
@@ -86,5 +86,5 @@ def demo_chain(input_text, memory):
86
  memory=memory
87
  )
88
 
89
- chat_reply = conversation.run(input=input_text)
90
  return chat_reply
 
52
  temperature = 0.1,repetition_penalty = 1.03)
53
 
54
  llm = HuggingFacePipeline.from_model_id(
55
+ #model_id =my_model_id,
56
  task="text-generation",
57
+ #model_kwargs={"trust_remote_code": True},
58
  pipeline_kwargs={
59
  "model": model,
60
  "tokenizer": tokenizer,
 
86
  memory=memory
87
  )
88
 
89
+ chat_reply = conversation.run(input=input_text, max_length = 50)
90
  return chat_reply