Update app.py

app.py

@@ -9,7 +9,8 @@ model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
 processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
 
 # Define a list of target words for the game
-words = ["cat", "car", "tree", "house", "dog"]  # Add more words as needed
+words = ["cat", "car", "tree", "house", "dog", "cloud", "flower", "bicycle", "boat", "star", "bird", "fish", "sun"]
+
 
 # Precompute text embeddings for faster comparisons
 text_inputs = processor(text=words, return_tensors="pt", padding=True)
@@ -53,7 +54,7 @@ interface = gr.Interface(
     inputs=gr.Sketchpad(),
     outputs="text",
     live=True,
-    description="Draw cat, car, tree, house, dog!"
+    description="Draw cat, car, tree, house, dog, cloud, flower, bicycle, boat, star, bird, fish, sun"
 )
 
 interface.launch()