dar-tau commited on
Commit
20c0832
1 Parent(s): bf7c081

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -73,6 +73,7 @@ def generate_interpretation_gpu(interpret_prompt, *args, **kwargs):
73
  return interpret_prompt.generate(*args, **kwargs)
74
 
75
 
 
76
  def run_interpretation(global_state, raw_interpretation_prompt, max_new_tokens, do_sample,
77
  temperature, top_k, top_p, repetition_penalty, length_penalty, use_gpu, i,
78
  num_beams=1):
 
73
  return interpret_prompt.generate(*args, **kwargs)
74
 
75
 
76
+ @spaces.GPU
77
  def run_interpretation(global_state, raw_interpretation_prompt, max_new_tokens, do_sample,
78
  temperature, top_k, top_p, repetition_penalty, length_penalty, use_gpu, i,
79
  num_beams=1):