Create app.py

app.py

@@ -0,0 +1,40 @@
+import gradio as gr
+from transformers import CLIPProcessor, CLIPModel
+from PIL import Image
+import torch
+
+# Load CLIP model and processor
+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"]  
+
+
+text_inputs = processor(text=words, return_tensors="pt", padding=True)
+with torch.no_grad():
+    text_features = model.get_text_features(**text_inputs)
+
+
+def guess_drawing(drawing):
+    image = Image.fromarray(drawing)  # Convert drawing to PIL image
+    image_inputs = processor(images=image, return_tensors="pt")
+    with torch.no_grad():
+        image_features = model.get_image_features(**image_inputs)
+
+    # Calculate cosine similarity with each word
+    similarity = torch.nn.functional.cosine_similarity(image_features, text_features)
+    best_match = words[similarity.argmax().item()]
+    return f"AI's guess: {best_match}"
+
+
+interface = gr.Interface(
+    fn=guess_drawing,
+    inputs=gr.Sketchpad(),
+    outputs="text",
+    live=True,
+    description="Draw something and see if the AI can guess it!"
+)
+
+interface.launch()
+