Update app.py
Browse files
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):
|