update reqs

app.py

@@ -8,13 +8,12 @@ from utils.audio import load_spectrograms
 from utils.compute_args import compute_args
 from utils.tokenize import tokenize, create_dict, sent_to_ix, cmumosei_2, cmumosei_7, pad_feature
 from model_LA import Model_LA
+import gradio as gr
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-working_dir = "."
-
 # load model
-ckpts_path = os.path.join(working_dir, 'ckpt')
+ckpts_path = 'ckpt'
 model_name = "Model_LA_e"
 # Listing sorted checkpoints
 ckpts = sorted(glob.glob(os.path.join(ckpts_path, model_name,'best*')), reverse=True)
@@ -63,7 +62,7 @@ def inference(video_path, text):
     pred = net(x, y, z).cpu().data.numpy()
     label_to_ix = ['happy', 'sad', 'angry', 'fear', 'disgust', 'surprise']
     result_dict = dict(zip(label_to_ix, pred[0]))
-    return out
+    return result_dict
 
 
 title="Emotion Recognition"

requirements.txt

@@ -1,2 +1,3 @@
 https://github.com/explosion/spacy-models/releases/download/en_vectors_web_lg-2.1.0/en_vectors_web_lg-2.1.0.tar.gz
-torch==1.9.1
+torch==1.9.1
+librosa