Spaces:
Running
Running
Peter
commited on
Commit
·
51de900
1
Parent(s):
b1deccc
support email links
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
|
|
5 |
import torch
|
6 |
from transformers import pipeline
|
7 |
|
8 |
-
from utils import postprocess, clear
|
9 |
|
10 |
logging.basicConfig(
|
11 |
level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
|
@@ -68,7 +68,7 @@ def generate_text(
|
|
68 |
if verbose:
|
69 |
logging.info(f"Generated text: {response}")
|
70 |
logging.info(f"Generation time: {rt:.2f}s")
|
71 |
-
return postprocess(response)
|
72 |
|
73 |
|
74 |
def get_parser():
|
@@ -153,10 +153,12 @@ if __name__ == "__main__":
|
|
153 |
label="Generated Result",
|
154 |
placeholder="The completed email will appear here",
|
155 |
)
|
156 |
-
|
157 |
-
|
158 |
-
|
159 |
-
|
|
|
|
|
160 |
|
161 |
gr.Markdown("## Advanced Options")
|
162 |
gr.Markdown(
|
@@ -208,7 +210,7 @@ if __name__ == "__main__":
|
|
208 |
no_repeat_ngram_size,
|
209 |
length_penalty,
|
210 |
],
|
211 |
-
outputs=[generated_email],
|
212 |
)
|
213 |
|
214 |
demo.launch(
|
|
|
5 |
import torch
|
6 |
from transformers import pipeline
|
7 |
|
8 |
+
from utils import postprocess, clear, make_email_link
|
9 |
|
10 |
logging.basicConfig(
|
11 |
level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
|
|
|
68 |
if verbose:
|
69 |
logging.info(f"Generated text: {response}")
|
70 |
logging.info(f"Generation time: {rt:.2f}s")
|
71 |
+
return postprocess(response), make_email_link(body=response)
|
72 |
|
73 |
|
74 |
def get_parser():
|
|
|
153 |
label="Generated Result",
|
154 |
placeholder="The completed email will appear here",
|
155 |
)
|
156 |
+
with gr.Row():
|
157 |
+
generate_button = gr.Button(
|
158 |
+
value="Generate!",
|
159 |
+
variant="primary",
|
160 |
+
)
|
161 |
+
email_link = gr.HTML("<p><em>A mailto: link will appear here</em></p>")
|
162 |
|
163 |
gr.Markdown("## Advanced Options")
|
164 |
gr.Markdown(
|
|
|
210 |
no_repeat_ngram_size,
|
211 |
length_penalty,
|
212 |
],
|
213 |
+
outputs=[generated_email, email_link],
|
214 |
)
|
215 |
|
216 |
demo.launch(
|