Try it

app.py

@@ -3,7 +3,7 @@ import os
 os.chdir('naacl-2021-fudge-controlled-generation/')
 
 import gradio as gr
-from predict_clickbait import generate_clickbait
+from fudge.predict_clickbait import generate_clickbait, tokenizer, classifier_tokenizer
 from datasets import load_dataset,DatasetDict,Dataset
 # from datasets import 
 from transformers import AutoTokenizer,AutoModelForSeq2SeqLM
@@ -13,13 +13,8 @@ import pandas as pd
 from sklearn.utils.class_weight import compute_class_weight
 import torch
 import pandas as pd 
-from model import Model
-import imp
+from fudge.model import Model
 import os
-import random
-import time
-import pickle
-import math
 from argparse import ArgumentParser
 from collections import namedtuple
 import mock
@@ -29,9 +24,8 @@ import numpy as np
 import torch.nn as nn
 import torch.nn.functional as F
 from data import Dataset
-from util import save_checkpoint, ProgressMeter, AverageMeter, num_params
-from constants import *
-from predict_clickbait import generate_clickbait, tokenizer, classifier_tokenizer
+from fudge.util import save_checkpoint, ProgressMeter, AverageMeter, num_params
+from fudge.constants import *
 
 
 # imp.reload(model)

{naacl-2021-fudge-controlled-generation → fudge}/requirements.txt

@@ -2,6 +2,9 @@ Phyme==0.0.9
 pronouncing==0.2.0
 pytorch-lightning==1.0.6
 torch==1.7.0
-tqdm==4.49.0
+tqdm==4.62.1
 sacrebleu==1.4.14
 sacremoses==0.0.43
+mock==4.0.3
+torchtext==0.8.1 --no-deps
+torchvision==0.8.2 --no-deps

requirements.txt

@@ -0,0 +1,10 @@
+Phyme==0.0.9
+pronouncing==0.2.0
+pytorch-lightning==1.0.6
+torch==1.7.0
+tqdm==4.62.1
+sacrebleu==1.4.14
+sacremoses==0.0.43
+mock==4.0.3
+torchtext==0.8.1 --no-deps
+torchvision==0.8.2 --no-deps