adding application

app.py

@@ -240,7 +240,7 @@ def find_last_token_index(full_ids, word_ids):
     return None
 
 @GPU  # this block runs on a job GPU
-def analyse_word(model_name: str, extraction_template: str, word: str, patchscopes_template: str):
+def analyse_word(model_name: str, word: str, patchscopes_template: str):
     try:
         model, tokenizer = get_model_and_tokenizer(model_name)
 
@@ -304,10 +304,6 @@ with gr.Blocks(theme="soft") as demo:
             choices=[DEFAULT_MODEL, "mistralai/Mistral-7B-v0.1", "meta-llama/Llama-2-7b", "Qwen/Qwen2-7B"],
             value=DEFAULT_MODEL,
         )
-        extraction_template = gr.Textbox(
-            label="Extraction prompt (use X as placeholder)",
-            value="repeat the following word X twice: 1)X 2)",
-        )
     patchscopes_template = gr.Textbox(
         label="Patchscopes prompt (use X as placeholder)",
         value="repeat the following word X twice: 1)X 2)",
@@ -318,7 +314,7 @@ with gr.Blocks(theme="soft") as demo:
 
     run_btn.click(
         analyse_word,
-        inputs=[model_name, extraction_template, word_box, patchscopes_template],
+        inputs=[model_name, word_box, patchscopes_template],
         outputs=out_html,
     )