updated utils

utils/utils.py

@@ -2,18 +2,14 @@ import matplotlib.pyplot as plt
 import numpy as np
 from utils.Caption import get_caption
 from utils.Emotions import get_emotions
-import spaces
 
-
-@spaces.GPU
 def get_label(image):
     caption = get_caption(image)
 
-    #pred = get_emotions(image, caption)
+    pred = get_emotions(image, caption)
 
     emotions = ['Excitement', 'Sadness', 'Amusement', 'Disgust', 'Awe', 'Contentment', 'Fear', 'Anger']
-    #probabilities = pred
-    probabilities = [10, 6, 8, 1, 5, 10, 6, 8]
+    probabilities = pred
     max_idx = np.argmax(probabilities)
 
     # Create color list where all bars are one color, and the max bar is another color