Update app.py

app.py

@@ -44,7 +44,6 @@ def get_mood_from_image(image: Image.Image):
     # Select the mood with the highest score
     selected_mood = max(filtered_moods, key=filtered_moods.get)
     return selected_mood
-
 def generate_art(mood):
     # Implement art generation logic using the Stable Diffusion API
     prompt = f"{mood} generative art"
@@ -68,11 +67,8 @@ def generate_art(mood):
 
     return image
 
-def mood_art_generator(image):
-    image_encoded = image.decode().split(",")[1]
-    image_decoded = base64.b64decode(image_encoded)
-    image_bytes = BytesIO(image_decoded)
-    image = Image.open(image_bytes)
+def mood_art_generator(image: np.ndarray):
+    image = Image.fromarray((image * 255).astype(np.uint8))
 
     mood = get_mood_from_image(image)
     if mood:
@@ -83,8 +79,8 @@ def mood_art_generator(image):
 
 iface = gr.Interface(
     fn=mood_art_generator,
-    inputs="image",
-    outputs="image",
+    inputs=gr.inputs.Image(shape=(224, 224), image_mode="RGB", source="upload"),
+    outputs=gr.outputs.Image(),
     title="Mood-based Art Generator",
     description="Upload an image of yourself and let the AI generate artwork based on your mood.",
     allow_flagging=False,
@@ -92,4 +88,4 @@ iface = gr.Interface(
     share=True
 )
 
-iface.launch()
+iface.launch()