update random url function

app.py

@@ -6,14 +6,10 @@ from PIL import Image
 from transformers import CLIPProcessor, CLIPModel
 from typing import List, Tuple
 
-pn.extension("texteditor", sizing_mode="stretch_width")
-
 
 def set_random_url(_):
-    if random.randint(0, 1) == 0:
-        api_url = "https://api.thecatapi.com/v1/images/search"
-    else:
-        api_url = "https://api.thedogapi.com/v1/images/search"
+    pet = random.choice(["cat", "dog"])
+    api_url = f"https://api.the{pet}api.com/v1/images/search"
     with requests.get(api_url) as resp:
         resp.raise_for_status()
         url = resp.json()[0]["url"]