Update app.py

app.py

@@ -12,9 +12,7 @@ description = """
 You can use this Space to test out the current model [intfloat/e5-mistral-7b-instruct](https://huggingface.co/intfloat/e5-mistral-7b-instruct). e5mistral has a larger context window, a different prompting/return mechanism and generally better results than other embedding models. 
 You can also use 🐣e5-mistral🛌🏻 by cloning this space. 🧬🔬🔍 Simply click here: <a style="display:inline-block" href="https://huggingface.co/spaces/Tonic/e5?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a></h3> 
 Join us : 🌟TeamTonic🌟 is always making cool demos! Join our active builder's🛠️community 👻  [![Join us on Discord](https://img.shields.io/discord/1109943800132010065?label=Discord&logo=discord&style=flat-square)](https://discord.gg/GWpVpekp) On 🤗Huggingface: [TeamTonic](https://huggingface.co/TeamTonic) & [MultiTransformer](https://huggingface.co/MultiTransformer) On 🌐Github: [Polytonic](https://github.com/tonic-ai) & contribute to 🌟 [Poly](https://github.com/tonic-ai/poly) 🤗Big thanks to Yuvi Sharma and all the folks at huggingface for the community grant 🤗
-You can use this space in **two ways !** either select an embeddings mode or 'None' to speak with the e5mistral LLM 🤗
 """
-
 os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:30'
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -49,18 +47,10 @@ def last_token_pool(last_hidden_states: Tensor, attention_mask: Tensor) -> Tenso
 
 @spaces.GPU
 def compute_embeddings(selected_task, input_text, system_prompt):
-    max_length = 2042  # Define max_length here
-
-    if selected_task == "None":
-        if system_prompt:
-            processed_texts = [f'Instruct: {system_prompt}\nQuery: {input_text}']
-        else:
-            processed_texts = [f'Query: {input_text}']
-    else:
-        task_description = tasks[selected_task]
-        processed_texts = [f'Instruct: {task_description}\nQuery: {input_text}']
+    max_length = 2042
+    task_description = tasks[selected_task]
+    processed_texts = [f'Instruct: {task_description}\nQuery: {input_text}']
 
-    task = tasks[selected_task]
     batch_dict = tokenizer(processed_texts, max_length=max_length - 1, return_attention_mask=False, padding=False, truncation=True)
     batch_dict['input_ids'] = [input_ids + [tokenizer.eos_token_id] for input_ids in batch_dict['input_ids']]
     batch_dict = tokenizer.pad(batch_dict, padding=True, return_attention_mask=True, return_tensors='pt')
@@ -76,8 +66,8 @@ def app_interface():
         gr.Markdown(title)
         gr.Markdown(description)
 
-        task_dropdown = gr.Dropdown(list(tasks.keys()) + ["None"], label="Select a Task (Optional)", value="None")
-        
+        task_dropdown = gr.Dropdown(list(tasks.keys()), label="Select a Task", value=list(tasks.keys())[0])
+     
         input_text_box = gr.Textbox(label="📖Input Text")
         system_prompt_box = gr.Textbox(label="🤖System Prompt (Optional)")
         
@@ -93,11 +83,12 @@ def app_interface():
                 compute_button
                 output_display
 
-        compute_button.click(
-            fn=compute_embeddings,
-            inputs=[task_dropdown, input_text_box, system_prompt_box], 
-            outputs=output_display
-        )
+    compute_button.click(
+        fn=compute_embeddings,
+        inputs=[task_dropdown, input_text_box],
+        outputs=output_display
+    )
+
 
     return demo