awitecki commited on
Commit
d704915
·
1 Parent(s): 7dfe0e4

Add missing imports

Browse files
Files changed (1) hide show
  1. app.py +8 -0
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)