update

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import CLIPProcessor, CLIPModel
 from PIL import Image
+import numpy as np
 import torch
 
 # Load CLIP model and processor
@@ -17,17 +18,28 @@ with torch.no_grad():
 
 
 def guess_drawing(drawing):
-    image = Image.fromarray(drawing)  # Convert drawing to PIL image
+   
+    drawing_data = drawing['data']  
+    image_array = np.array(drawing_data, dtype=np.uint8)  
+    
+    
+    image = Image.fromarray(image_array)
+    
+    
     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 the AI's best guess
     return f"AI's guess: {best_match}"
 
-
+# Set up Gradio interface
 interface = gr.Interface(
     fn=guess_drawing,
     inputs=gr.Sketchpad(),