Update app.py

app.py

@@ -172,67 +172,70 @@ def generate_image(prompt):
         ]
 
         if "face" in prompt:
-        output = random.choice(faces)
-        return output
-    elif "landscape" in prompt:
-        output = random.choice(landscapes)
-        return output
-    elif "animal" in prompt:
-        output = random.choice(animal)
-        return output
-    elif "car" in prompt:
-        output = random.choice(car)
-        return output
-    elif "food" in prompt:
-        output = random.choice(food)
-        return output
-    elif "boat" in prompt:
-        output = random.choice(boat)
-        return output
-    elif "logo" in prompt:
-        output = random.choice(logo)
-        return output
-    elif "eye" in prompt:
-        output = random.choice(eye)
-        return output
-    elif "beard" in prompt:
-        output = random.choice(beard)
-        return output
-    elif "avatar" in prompt:
-        output = random.choice(avatar)
-        return output
-    elif "dreadlocks" in prompt:
-        output = random.choice(dreadlocks)
-        return output
-    elif "soap" in prompt:
-        output = random.choice(soap)
-        return output
-    elif "sumsung" in prompt:
-        output = random.choice(sumsung)
-        return output
-    elif "iphone" in prompt:
-        output = random.choice(iphone)
-        return output
-    elif "phone" in prompt:
-        output = random.choice(phone)
-        return output
-    elif "code" in prompt:
-        output = random.choice(code)
-        return output
-    elif "youtube" in prompt:
-        output = random.choice(ytlogo)
-        return output
-    elif "dog" in prompt:
-        output = random.choice(dog)
-        return output
-    elif "doll" in prompt:
-        output = random.choice(doll)
-        return output
-    elif "background" in prompt:
-        output = random.choice(background)
-        return output
-    else:
-        return "No image generated."
+            output = random.choice(faces)
+            return output
+        elif "landscape" in prompt:
+            output = random.choice(landscapes)
+            return output
+        elif "animal" in prompt:
+            output = random.choice(animal)
+            return output
+        elif "car" in prompt:
+            output = random.choice(car)
+            return output
+        elif "food" in prompt:
+            output = random.choice(food)
+            return output
+        elif "boat" in prompt:
+            output = random.choice(boat)
+            return output
+        elif "logo" in prompt:
+            output = random.choice(logo)
+            return output
+        elif "eye" in prompt:
+            output = random.choice(eye)
+            return output
+        elif "food" in prompt:
+            output = random.choice(food)
+            return output
+        elif "beard" in prompt:
+            output = random.choice(beard)
+            return output
+        elif "avatar" in prompt:
+            output = random.choice(avatar)
+            return output
+        elif "dreadlocks" in prompt:
+            output = random.choice(dreadlocks)
+            return output
+        elif "soap" in prompt:
+            output = random.choice(soap)
+            return output
+        elif "sumsung" in prompt:
+            output = random.choice(sumsung)
+            return output
+        elif "iphone" in prompt:
+            output = random.choice(iphone)
+            return output
+        elif "phone" in prompt:
+            output = random.choice(phone)
+            return output
+        elif "code" in prompt:
+            output = random.choice(code)
+            return output
+        elif "youtube" in prompt:
+            output = random.choice(ytlogo)
+            return output
+        elif "dog" in prompt:
+            output = random.choice(dog)
+            return output
+        elif "doll" in prompt:
+            output = random.choice(doll)
+            return output
+        elif "background" in prompt:
+            output = random.choice(background)
+            return output
+        else:
+            return "No image generated."
 
 iface = gr.Interface(
     fn=generate_image,