Update app.py

app.py

@@ -69,7 +69,7 @@ def reset_model(model_name, *extra_components, with_extra_components=True):
                 + [gr.Button('', visible=False) for _ in range(len(tokens_container))]
                 + [*extra_components])
     
-
+@spaces.GPU
 def get_hidden_states(raw_original_prompt):
     model, tokenizer = global_state.model, global_state.tokenizer
     original_prompt = global_state.original_prompt_template.format(prompt=raw_original_prompt)