Spaces:
Runtime error
Runtime error
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(
|