Add missing imports
Browse files
app.py
CHANGED
@@ -4,6 +4,14 @@ from fastai.vision.all import *
|
|
4 |
import gradio as gr
|
5 |
from blurr.text.data.all import *
|
6 |
from blurr.text.modeling.all import *
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
|
8 |
pre_trained_model_name = "sshleifer/distilbart-cnn-6-6"
|
9 |
hf_arch, hf_config, hf_tokenizer, hf_model = BlurrText().get_hf_objects(pre_trained_model_name, model_cls=BartForConditionalGeneration)
|
|
|
4 |
import gradio as gr
|
5 |
from blurr.text.data.all import *
|
6 |
from blurr.text.modeling.all import *
|
7 |
+
from fastai.data.all import *
|
8 |
+
from fastai.callback.all import *
|
9 |
+
from fastai.learner import *
|
10 |
+
from fastai.optimizer import *
|
11 |
+
from transformers import *
|
12 |
+
import nltk
|
13 |
+
|
14 |
+
nltk.download('punkt', quiet=True)
|
15 |
|
16 |
pre_trained_model_name = "sshleifer/distilbart-cnn-6-6"
|
17 |
hf_arch, hf_config, hf_tokenizer, hf_model = BlurrText().get_hf_objects(pre_trained_model_name, model_cls=BartForConditionalGeneration)
|