Update app.py

app.py

@@ -44,7 +44,7 @@ ATTRIBUTES_DICT = attributes_data['attribute_mapping']
 
 def shot(input, category, level):
     output_dict = {}
-    if level = 'variant':
+    if level == 'variant':
         subColour,mainColour,score = get_colour(ast.literal_eval(str(input)),category)    
         openai_parsed_response = get_openAI_tags(ast.literal_eval(str(input)))
         face_embeddings = get_face_embeddings(ast.literal_eval(str(input)))
@@ -59,7 +59,7 @@ def shot(input, category, level):
         output_dict['cropped_images'] = cropped_images
 
 
-    if level = 'product':
+    if level == 'product':
         common_result = get_predicted_attributes(ast.literal_eval(str(input)),category)    
         output_dict['attributes'] = common_result
         output_dict['subcategory'] = category