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