Spaces:
Sleeping
Sleeping
Update helper.py
Browse files
helper.py
CHANGED
@@ -146,7 +146,7 @@ def registry(name: str = None, **kwargs) -> gr.ChatInterface:
|
|
146 |
_, _, preprocess, postprocess = get_interface_args("chat")
|
147 |
|
148 |
# Get the predict function
|
149 |
-
predict_fn = get_fn(
|
150 |
|
151 |
# Define a wrapper function that integrates preprocessing and postprocessing
|
152 |
def wrapper(message, history, system_prompt, temperature, max_tokens, top_k, repetition_penalty, top_p):
|
|
|
146 |
_, _, preprocess, postprocess = get_interface_args("chat")
|
147 |
|
148 |
# Get the predict function
|
149 |
+
predict_fn = get_fn(model_name=name, **kwargs)
|
150 |
|
151 |
# Define a wrapper function that integrates preprocessing and postprocessing
|
152 |
def wrapper(message, history, system_prompt, temperature, max_tokens, top_k, repetition_penalty, top_p):
|