Update app.py

app.py

@@ -97,21 +97,15 @@ async def process_file(file_data: FileProcess):
 
 
 def process_video(video_path):
-    # 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']}
+    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
 
 
 if __name__ == "__main__":