rodrigomasini commited on
Commit
6bf705e
·
verified ·
1 Parent(s): 543fed2

Update helper.py

Browse files
Files changed (1) hide show
  1. helper.py +1 -1
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(model_path=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):
 
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):