Update app.py
Browse files
app.py
CHANGED
@@ -39,6 +39,10 @@ suggested_interpretation_prompts = ["Before responding, let me repeat the messag
|
|
39 |
|
40 |
|
41 |
## functions
|
|
|
|
|
|
|
|
|
42 |
def get_hidden_states(raw_original_prompt):
|
43 |
original_prompt = original_prompt_template.format(prompt=raw_original_prompt)
|
44 |
model_inputs = tokenizer(original_prompt, add_special_tokens=False, return_tensors="pt").to(model.device)
|
|
|
39 |
|
40 |
|
41 |
## functions
|
42 |
+
@spaces.GPU
|
43 |
+
def initialize_gpu():
|
44 |
+
pass
|
45 |
+
|
46 |
def get_hidden_states(raw_original_prompt):
|
47 |
original_prompt = original_prompt_template.format(prompt=raw_original_prompt)
|
48 |
model_inputs = tokenizer(original_prompt, add_special_tokens=False, return_tensors="pt").to(model.device)
|