schroneko commited on
Commit
ac793de
·
verified ·
1 Parent(s): 3b9151f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -11,7 +11,9 @@ logging.basicConfig(level=logging.INFO)
11
 
12
 
13
  def sanitize_filename(prompt):
14
- sanitized = re.sub(r"[^\w\d-]", "_", prompt)
 
 
15
  return sanitized[:50]
16
 
17
 
 
11
 
12
 
13
  def sanitize_filename(prompt):
14
+ sanitized = re.sub(r'[^\w\-]', '_', prompt)
15
+ sanitized = re.sub(r'_+', '_', sanitized)
16
+ sanitized = sanitized.strip('_')
17
  return sanitized[:50]
18
 
19