Update chatbot.py
Browse files- 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
|