Changed path to models

app.py

@@ -182,13 +182,13 @@ decision_frameworks = {
 def predict(video_file, video_model_name, audio_model_name, framework_name):
 
     image_processor = VivitImageProcessor.from_pretrained("google/vivit-b-16x2-kinetics400")
-    video_model = torch.load(video_model_name)
+    video_model = torch.load('./' + video_model_name)
 
     model_id = "facebook/wav2vec2-large"
     config = AutoConfig.from_pretrained(model_id, num_labels=6)
     audio_processor = AutoFeatureExtractor.from_pretrained(model_id)
     audio_model = Wav2Vec2ForSequenceClassification.from_pretrained(model_id, config=config)
-    audio_model.load_state_dict(torch.load(audio_model_name))
+    audio_model.load_state_dict(torch.load('./' + audio_model_name))
     audio_model.eval()
 
     delete_directory_path = "./temp/"