adding application

app.py

@@ -126,7 +126,6 @@ class PatchscopesRetriever(WordRetrieverBase):
 
         return last_token_hidden_states
 
-    @GPU  # this block runs on a job GPU
     def get_hidden_states_and_retrieve_word(self, word, num_tokens_to_generate=None):
         last_token_hidden_states = self.extract_hidden_states(word)
         patchscopes_description_by_layers = self.retrieve_word(
@@ -240,7 +239,7 @@ def find_last_token_index(full_ids, word_ids):
             return i + len(word_ids) - 1
     return None
 
-
+@GPU  # this block runs on a job GPU
 def analyse_word(model_name: str, extraction_template: str, word: str, patchscopes_template: str):
     try:
         model, tokenizer = get_model_and_tokenizer(model_name)