Tonic commited on
Commit
b85ac58
·
verified ·
1 Parent(s): 8421fc1

Update app.py

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