Update app.py

app.py

@@ -11,22 +11,18 @@ with zipfile.ZipFile("images/stable_diffusion.zip","r") as zip_ref:
     zip_ref.extractall("images/")
    
 def open_stable_ims(profession):
-    if profession and len(profession) != 0:
+    if len(profession) != 0:
         dirname = 'images/stable_diffusion/'+ profession+'/'
-        if os.path.exists(dirname):
-            images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
-            random.shuffle(images)
-            return images[:16]
-    return []
+        images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
+        random.shuffle(images)
+        return images[:16]
 
 def open_fair_ims(profession):
-    if profession and len(profession) != 0:
+    if len(profession) != 0:
         dirname = 'images/fair_diffusion/' + profession+'/'
-        if os.path.exists(dirname):
-            images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
-            random.shuffle(images)
-            return images[:16]
-    return []
+        images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
+        random.shuffle(images)
+        return images[:16]
 
 professions = sorted(os.listdir('images/fair_diffusion'))