dar-tau commited on
Commit
1a614f9
1 Parent(s): 9dd96f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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)