d3dname commited on
Commit
04e18e4
·
verified ·
1 Parent(s): 084f923

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -39,6 +39,8 @@ def generate_email(company, recipient_name, product, progress=gr.Progress()):
39
  full_response = result['response'][0]
40
  # Extract the email content after [/INST]
41
  email_content = full_response.split('[/INST]')[-1].strip()
 
 
42
  # Decode Unicode escape sequences
43
  email_content = decode_unicode_escape(email_content)
44
  result_queue.put(("success", email_content))
 
39
  full_response = result['response'][0]
40
  # Extract the email content after [/INST]
41
  email_content = full_response.split('[/INST]')[-1].strip()
42
+
43
+ email_content = email_content.split('Subject:')[-1].strip()
44
  # Decode Unicode escape sequences
45
  email_content = decode_unicode_escape(email_content)
46
  result_queue.put(("success", email_content))