Update app.py

app.py

@@ -57,7 +57,7 @@ def predict(image):
 def save_feedback(image, correct_label):
     global feedback_data
     try:
-        image_np = np.array(image['composite']).astype('uint8')
+        image_np = np.array(image).astype('uint8')
         image_pil = Image.fromarray(image_np, 'RGBA').convert('RGB')
         image_path = f"images/{len(feedback_data)}.png"
         image_pil.save(image_path)
@@ -86,7 +86,7 @@ def retrain_model(feedback_data):
 
         dataset_dict = {
             "image": [load_image(f) for f in feedback_data["image_path"]],
-            "label": feedback_data["correct_label"].tolist()
+            "label": feedback_data["correct_label"].astype(int).tolist()  # Convert labels to integers
         }
 
         dataset = Dataset.from_dict(dataset_dict)