Dusan Svilarkovic commited on
Commit
6b337ea
1 Parent(s): fcf9373

Fixing app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  import gradio as gr
2
  from predict_clickbait import generate_clickbait
3
  from datasets import load_dataset,DatasetDict,Dataset
@@ -28,9 +32,7 @@ from data import Dataset
28
  from util import save_checkpoint, ProgressMeter, AverageMeter, num_params
29
  from constants import *
30
  from predict_clickbait import generate_clickbait, tokenizer, classifier_tokenizer
31
- import os
32
 
33
- os.chdir('naacl-2021-fudge-controlled-generation/')
34
 
35
  # imp.reload(model)
36
  pretrained_model = "../checkpoint-150/"
 
1
+ import os
2
+
3
+ os.chdir('naacl-2021-fudge-controlled-generation/')
4
+
5
  import gradio as gr
6
  from predict_clickbait import generate_clickbait
7
  from datasets import load_dataset,DatasetDict,Dataset
 
32
  from util import save_checkpoint, ProgressMeter, AverageMeter, num_params
33
  from constants import *
34
  from predict_clickbait import generate_clickbait, tokenizer, classifier_tokenizer
 
35
 
 
36
 
37
  # imp.reload(model)
38
  pretrained_model = "../checkpoint-150/"