Tonic commited on
Commit
b608d51
Β·
verified Β·
1 Parent(s): 3c37c5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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