Update app.py

app.py

@@ -97,16 +97,21 @@ async def process_file(file_data: FileProcess):
 
 
 def process_video(video_path):
-    learn = load_learner('model.pkl')
-    searches = ['formal','casual','athletic']
-    searches = sorted(searches) # need to be sorted because that's the order the predictions will be outputed
-    values = [i for i in range(0,len(searches))]
-    class_dict = dict(zip(searches,values))
-
-    classication,_,probs = learn.predict(im)
-    confidences = {label: float(probs[i]) for i, label in enumerate(class_dict)} 
-    return confidences
-
+    # Load the image from the provided path
+    img = PILImage.create(video_path)
+    
+    # Make the prediction
+    classification, _, probs = learn.predict(img)
+    
+    # Convert the prediction to a confidence dictionary
+    confidences = {label: float(probs[i]) for i, label in enumerate(class_dict)}
+    
+    # If classification is not formal, return 'informal'
+    if classification != 'formal':
+        informal_confidence = sum(confidences[label] for label in class_dict if label != 'formal')
+        return {'informal': informal_confidence}
+    else:
+        return {'formal': confidences['formal']}
 
 if __name__ == "__main__":
     logger.info("Starting the Face Emotion Recognition API")