Spaces:
Runtime error
Runtime error
Commit
·
06cc40d
1
Parent(s):
de379ac
Update app.py
Browse files
app.py
CHANGED
|
@@ -237,62 +237,62 @@ def generate_image(prompt):
|
|
| 237 |
]
|
| 238 |
|
| 239 |
|
| 240 |
-
|
| 241 |
-
|
| 242 |
-
|
| 243 |
-
|
| 244 |
-
|
| 245 |
-
|
| 246 |
-
|
| 247 |
-
|
| 248 |
-
|
| 249 |
-
|
| 250 |
-
|
| 251 |
-
|
| 252 |
-
|
| 253 |
-
|
| 254 |
-
|
| 255 |
-
|
| 256 |
-
|
| 257 |
-
|
| 258 |
-
|
| 259 |
-
|
| 260 |
-
|
| 261 |
-
|
| 262 |
-
|
| 263 |
-
|
| 264 |
-
|
| 265 |
-
|
| 266 |
-
|
| 267 |
-
|
| 268 |
-
|
| 269 |
-
|
| 270 |
-
|
| 271 |
-
|
| 272 |
-
|
| 273 |
-
|
| 274 |
-
|
| 275 |
-
|
| 276 |
-
|
| 277 |
-
|
| 278 |
-
|
| 279 |
-
|
| 280 |
-
|
| 281 |
-
|
| 282 |
-
|
| 283 |
-
|
| 284 |
-
|
| 285 |
-
|
| 286 |
-
|
| 287 |
-
|
| 288 |
-
|
| 289 |
-
|
| 290 |
-
|
| 291 |
-
|
| 292 |
-
|
| 293 |
-
|
| 294 |
-
|
| 295 |
-
|
| 296 |
|
| 297 |
image = load_image(output_url)
|
| 298 |
return image
|
|
|
|
| 237 |
]
|
| 238 |
|
| 239 |
|
| 240 |
+
if "face" in prompt:
|
| 241 |
+
output_url = random.choice(faces)
|
| 242 |
+
elif "landscape" in prompt:
|
| 243 |
+
output_url = random.choice(landscapes)
|
| 244 |
+
elif "animal" in prompt:
|
| 245 |
+
output_url = random.choice(animal)
|
| 246 |
+
elif "car" in prompt:
|
| 247 |
+
output_url = random.choice(car)
|
| 248 |
+
elif "food" in prompt:
|
| 249 |
+
output_url = random.choice(food)
|
| 250 |
+
elif "boat" in prompt:
|
| 251 |
+
output_url = random.choice(boat)
|
| 252 |
+
elif "logo" in prompt:
|
| 253 |
+
output_url = random.choice(logo)
|
| 254 |
+
elif "eye" in prompt:
|
| 255 |
+
output_url = random.choice(eye)
|
| 256 |
+
elif "beard" in prompt:
|
| 257 |
+
output_url = random.choice(beard)
|
| 258 |
+
elif "avatar" in prompt:
|
| 259 |
+
output_url = random.choice(avatar)
|
| 260 |
+
elif "dreadlocks" in prompt:
|
| 261 |
+
output_url = random.choice(dreadlocks)
|
| 262 |
+
elif "soap" in prompt:
|
| 263 |
+
output_url = random.choice(soap)
|
| 264 |
+
elif "sumsung" in prompt:
|
| 265 |
+
output_url = random.choice(sumsung)
|
| 266 |
+
elif "iphone" in prompt:
|
| 267 |
+
output_url = random.choice(iphone)
|
| 268 |
+
elif "phone" in prompt:
|
| 269 |
+
output_url = random.choice(phone)
|
| 270 |
+
elif "code" in prompt:
|
| 271 |
+
output_url = random.choice(code)
|
| 272 |
+
elif "youtube" in prompt:
|
| 273 |
+
output_url = random.choice(ytlogo)
|
| 274 |
+
elif "dog" in prompt:
|
| 275 |
+
output_url = random.choice(dog)
|
| 276 |
+
elif "doll" in prompt:
|
| 277 |
+
output_url = random.choice(doll)
|
| 278 |
+
elif "background" in prompt:
|
| 279 |
+
output_url = random.choice(background)
|
| 280 |
+
elif "rock" in prompt:
|
| 281 |
+
output_url = random.choice(rocks)
|
| 282 |
+
elif "gold robot" in prompt:
|
| 283 |
+
output_url = random.choice(goldrobot)
|
| 284 |
+
elif "house" in prompt:
|
| 285 |
+
output_url = random.choice(house)
|
| 286 |
+
elif "horse" in prompt:
|
| 287 |
+
output_url = random.choice(horse)
|
| 288 |
+
elif "money" in prompt:
|
| 289 |
+
output_url = random.choice(money)
|
| 290 |
+
elif "rainbow" in prompt:
|
| 291 |
+
output_url = random.choice(rainbow)
|
| 292 |
+
elif "rain" in prompt:
|
| 293 |
+
output_url = random.choice(rain)
|
| 294 |
+
else:
|
| 295 |
+
return "No image generated."
|
| 296 |
|
| 297 |
image = load_image(output_url)
|
| 298 |
return image
|