BaggerOfWords commited on
Commit
7396944
·
1 Parent(s): fdf4266

issue with spaces decorators

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -105,8 +105,8 @@ def apply_config2():
105
  # (n_models_state, *all textboxes, *all load buttons, *all status boxes)
106
  return (n_vis, *tb_updates, *btn_updates, *status_updates)
107
 
108
- @spaces.GPU()
109
  # Load a single model and report status
 
110
  def load_single_model(model_path, use_bfloat16=True):
111
  try:
112
  repo = model_path
@@ -135,6 +135,7 @@ def check_all_loaded(n_visible, *status_texts):
135
  return gr.update(interactive=True)
136
  return gr.update(interactive=False)
137
 
 
138
  def run_scoring(input_text, *args):
139
  """
140
  args: first MAX_MODELS entries are model paths, followed by threshold_choice and custom_threshold
 
105
  # (n_models_state, *all textboxes, *all load buttons, *all status boxes)
106
  return (n_vis, *tb_updates, *btn_updates, *status_updates)
107
 
 
108
  # Load a single model and report status
109
+ @spaces.GPU()
110
  def load_single_model(model_path, use_bfloat16=True):
111
  try:
112
  repo = model_path
 
135
  return gr.update(interactive=True)
136
  return gr.update(interactive=False)
137
 
138
+ @spaces.GPU()
139
  def run_scoring(input_text, *args):
140
  """
141
  args: first MAX_MODELS entries are model paths, followed by threshold_choice and custom_threshold