Spanicin commited on
Commit
470dd9c
·
verified ·
1 Parent(s): 0b9087e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -280,6 +280,11 @@ def run_preprocessing(args):
280
 
281
  return preprocessed_data
282
 
 
 
 
 
 
283
  @app.route("/run", methods=['POST'])
284
  def generate_video():
285
  global start_time
@@ -333,6 +338,7 @@ def generate_video():
333
  try:
334
  data = json.loads(json_data)
335
  text_prompt = data.get('content')
 
336
  app.config['text_prompt'] = text_prompt
337
  end_time_ryze = time.time()
338
  diff = end_time_ryze - start_time_ryze
 
280
 
281
  return preprocessed_data
282
 
283
+ def remove_brackets(text):
284
+ # Use regex to remove content in brackets at the end of the text
285
+ cleaned_text = re.sub(r'\s*\[.*?\]\s*$', '', text)
286
+ return cleaned_text.strip()
287
+
288
  @app.route("/run", methods=['POST'])
289
  def generate_video():
290
  global start_time
 
338
  try:
339
  data = json.loads(json_data)
340
  text_prompt = data.get('content')
341
+ text_prompt = remove_brackets(text_prompt)
342
  app.config['text_prompt'] = text_prompt
343
  end_time_ryze = time.time()
344
  diff = end_time_ryze - start_time_ryze