Update app.py

app.py

@@ -69,28 +69,24 @@ def generate_art(mood):
     return image
 
 def mood_art_generator(image):
-    # Convert base64 string to PIL image
     image_encoded = image.split(",")[1]
-    image = Image.open(BytesIO(base64.b64decode(image_encoded)))
-
-    mood = get_mood_from_image(np.array(image))
+    # Decode the base64 encoded image
+    image_decoded = base64.b64decode(image_encoded)
+    # Convert the image to a PIL Image
+    image = Image.open(BytesIO(image_decoded)).convert("RGB")
+    mood = get_mood_from_image(image)
     if mood:
         art = generate_art(mood)
         return art
     else:
         return None
 
-image_input = gr.inputs.Image(shape=(224, 224))
-outputs = gr.outputs.Image(type = "pil", label="Generated Artwork")
+image_input = gr.inputs.Image(shape=(224, 224), image_mode="RGB", source="upload")
+outputs = gr.outputs.Image(type="pil", label="Generated Artwork")
 
 interface = gr.Interface(
     fn=mood_art_generator,
     inputs=image_input,
     outputs=outputs,
     title="Mood-based Art Generator",
-    description="Upload an image of yourself and let the AI generate artwork based on your mood.",
-)
-
-if __name__ == "__main__":
-    interface.launch()
-
+    description="Upload an image of yourself and let the AI