1inkusFace commited on
Commit
cb29eaf
Β·
verified Β·
1 Parent(s): cbc782e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -307,14 +307,14 @@ def expand_prompt(prompt):
307
  enhanced_prompt = txt_tokenizer.decode(outputs[0], skip_special_tokens=True)
308
  print('-- generated prompt 1 --')
309
  print(enhanced_prompt)
310
- enhanced_prompt = filter_text(enhanced_prompt,prompt)
311
- enhanced_prompt = filter_text(enhanced_prompt,user_prompt_rewrite)
312
- enhanced_prompt = filter_text(enhanced_prompt,system_prompt_rewrite)
313
  print('-- filtered prompt --')
314
- target_phrase = "The rewritten prompt:"
315
- pattern = r".*(?=" + re.escape(target_phrase) + r")"
316
- enhanced_prompt = re.sub(pattern, "", enhanced_prompt)
317
- filter_text(enhanced_prompt,target_phrase)
318
  print(enhanced_prompt)
319
  '''
320
  input_text_2 = f"{system_prompt_rewrite} {user_prompt_rewrite_2} {enhanced_prompt}"
 
307
  enhanced_prompt = txt_tokenizer.decode(outputs[0], skip_special_tokens=True)
308
  print('-- generated prompt 1 --')
309
  print(enhanced_prompt)
310
+ #enhanced_prompt = filter_text(enhanced_prompt,prompt)
311
+ #enhanced_prompt = filter_text(enhanced_prompt,user_prompt_rewrite)
312
+ #enhanced_prompt = filter_text(enhanced_prompt,system_prompt_rewrite)
313
  print('-- filtered prompt --')
314
+ #target_phrase = "The rewritten prompt:"
315
+ #pattern = r".*(?=" + re.escape(target_phrase) + r")"
316
+ #enhanced_prompt = re.sub(pattern, "", enhanced_prompt)
317
+ #filter_text(enhanced_prompt,target_phrase)
318
  print(enhanced_prompt)
319
  '''
320
  input_text_2 = f"{system_prompt_rewrite} {user_prompt_rewrite_2} {enhanced_prompt}"