yabramuvdi commited on
Commit
0c91381
·
verified ·
1 Parent(s): 310d018

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,7 +79,7 @@ def predict_next_token(model_name, text, top_k, custom_token=""):
79
  def append_selected_token(text, selected_token):
80
  """Append selected token from dropdown to the text input."""
81
  if selected_token:
82
- text += f" {selected_token.strip('\'')}"
83
  return text
84
 
85
  # Create the UI
 
79
  def append_selected_token(text, selected_token):
80
  """Append selected token from dropdown to the text input."""
81
  if selected_token:
82
+ text += f" {selected_token.strip(\"'\")}"
83
  return text
84
 
85
  # Create the UI