Update app.py

app.py

@@ -69,18 +69,19 @@ def generate_art(mood):
     return image
 
 def mood_art_generator(image):
-    # Convert the base64 encoded image to a PIL Image
-    img = Image.open(BytesIO(base64.b64decode(image.split(",")[1])))
+    # Convert base64 string to PIL image
+    image_encoded = image.split(",")[1]
+    image = Image.open(BytesIO(base64.b64decode(image_encoded)))
 
-    mood = get_mood_from_image(img)
+    mood = get_mood_from_image(np.array(image))
     if mood:
         art = generate_art(mood)
         return art
     else:
         return None
 
-image_input = gr.inputs.Image(shape=(224, 224), source="upload")
-outputs = gr.outputs.Image(type = "pil",label="Generated Artwork")
+image_input = gr.inputs.Image(shape=(224, 224))
+outputs = gr.outputs.Image(type = "pil", label="Generated Artwork")
 
 interface = gr.Interface(
     fn=mood_art_generator,
@@ -92,3 +93,4 @@ interface = gr.Interface(
 
 if __name__ == "__main__":
     interface.launch()
+