Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -50,6 +50,7 @@ def last_token_pool(last_hidden_states: Tensor, attention_mask: Tensor) -> Tenso
|
|
50 |
|
51 |
@spaces.GPU
|
52 |
def compute_embeddings(selected_task, input_text, system_prompt):
|
|
|
53 |
|
54 |
if selected_task == "None":
|
55 |
# Use the system prompt if 'None' is selected
|
|
|
50 |
|
51 |
@spaces.GPU
|
52 |
def compute_embeddings(selected_task, input_text, system_prompt):
|
53 |
+
max_length = 2042 # Define max_length here
|
54 |
|
55 |
if selected_task == "None":
|
56 |
# Use the system prompt if 'None' is selected
|