Commit
•
dd4459d
1
Parent(s):
b40d8e7
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,6 +2,8 @@ import gradio as gr
|
|
| 2 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 3 |
import nltk
|
| 4 |
nltk.download('punkt')
|
|
|
|
|
|
|
| 5 |
|
| 6 |
def generate_subject(email_body):
|
| 7 |
model_name = 'anukvma/t5-base-medium-email-subject-generation-v2'
|
|
|
|
| 2 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 3 |
import nltk
|
| 4 |
nltk.download('punkt')
|
| 5 |
+
nltk.download('wordnet')
|
| 6 |
+
nltk.download('omw-1.4')
|
| 7 |
|
| 8 |
def generate_subject(email_body):
|
| 9 |
model_name = 'anukvma/t5-base-medium-email-subject-generation-v2'
|