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)
|