Update chatbot.py
Browse files- chatbot.py +1 -2
chatbot.py
CHANGED
@@ -22,8 +22,7 @@ def load_model():
|
|
22 |
|
23 |
@st.cache_resource
|
24 |
def load_pipeline():
|
25 |
-
|
26 |
-
max_tokens = 50,top_k = 30, temperature = 0.1,repetition_penalty = 1.03)
|
27 |
hf = HuggingFacePipeline(pipeline=pipe)
|
28 |
return hf
|
29 |
|
|
|
22 |
|
23 |
@st.cache_resource
|
24 |
def load_pipeline():
|
25 |
+
pipe = pipeline("text-generation", model="KvrParaskevi/Llama-2-7b-Hotel-Booking-Model", max_tokens = 50,top_k = 30, temperature = 0.1,repetition_penalty = 1.03)
|
|
|
26 |
hf = HuggingFacePipeline(pipeline=pipe)
|
27 |
return hf
|
28 |
|