Update app.py

app.py

@@ -3,6 +3,7 @@ from PIL import Image
 import requests
 from io import BytesIO
 import random
+import time
 
 def load_image(url):
     response = requests.get(url)
@@ -180,6 +181,14 @@ def generate_image(prompt):
           "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcR7i4xPXauC7fC6R4gz8TgcQCPkqwAg2olTPDgSOFm3i3FLLxD8Xf9uFg5BZOs&s"
         ]
 
+        rocks = [
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcT3PhKY8mj-GRrwr3wZbpZq_QOi4PPff8XhQ3NMCP66xESicYz6IcOlXNVtMzk&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcRRigOBUBYtxsOa3dz3uc8n6Px8BNTWgwurbO3DRbMETydwWJ_oG8likIwP2g&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSw9BesCM_YHs4Iou8-fSEVAI8Qz_dCfKBRSDzy2GjzSz2biE2BBMkOM_o8Ow&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSkpCg5fj_yKPuxi-YG3-q8IuVq66Rsw_3dJHgCJTVlJlU2MociE1zvLz0FuOg&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSx9XrO9FSPBsrAmm4X5Pz92ngVxx3wBrtuxQqwHsk_9XpPuOP5y23eVvAtJw&s"
+        ]
+
         if "face" in prompt:
             output_url = random.choice(faces)
         elif "landscape" in prompt:
@@ -220,6 +229,8 @@ def generate_image(prompt):
             output_url = random.choice(doll)
         elif "background" in prompt:
             output_url = random.choice(background)
+        elif "rock" in prompt:
+            output_url = random.choice(rocks)
         else:
             return "No image generated."