Update app.py

app.py

@@ -49,12 +49,15 @@ def last_token_pool(last_hidden_states: Tensor, attention_mask: Tensor) -> Tenso
 
 @spaces.GPU
 def compute_embeddings(selected_task, *input_texts):
-    max_length = 2042
-    if selected_task:
-        task = tasks[selected_task]
-        processed_texts = [f'Instruct: {task}\nQuery: {text}' for text in input_texts]
+
+    if selected_task == "None":
+        # Use the system prompt if 'None' is selected
+        processed_texts = [f'Instruct: {system_prompt}\nQuery: {input_text}']
     else:
-        processed_texts = [f'Instruct: {system_prompt}\nQuerry: {text}' for text in input_texts]
+        # Use the task description from the tasks dictionary
+        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']]