Update app.py

app.py

@@ -48,7 +48,7 @@ def last_token_pool(last_hidden_states: Tensor, attention_mask: Tensor) -> Tenso
         return last_hidden_states[torch.arange(batch_size, device=last_hidden_states.device), sequence_lengths]
 
 @spaces.GPU
-def compute_embeddings(selected_task, *input_texts):
+def compute_embeddings(selected_task, input_text, system_prompt):
 
     if selected_task == "None":
         # Use the system prompt if 'None' is selected