Kyan14 commited on
Commit
de54af9
·
1 Parent(s): c86dde0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -9
app.py CHANGED
@@ -69,24 +69,22 @@ def generate_art(mood):
69
  return image
70
 
71
  def mood_art_generator(image):
72
- image_encoded = image.split(",")[1]
73
- # Decode the base64 encoded image
74
- image_decoded = base64.b64decode(image_encoded)
75
- # Convert the image to a PIL Image
76
- image = Image.open(BytesIO(image_decoded)).convert("RGB")
77
- mood = get_mood_from_image(image)
78
  if mood:
79
  art = generate_art(mood)
80
  return art
81
  else:
82
  return None
83
 
84
- image_input = gr.inputs.Image(shape=(224, 224), image_mode="RGB", source="upload")
85
- outputs = gr.outputs.Image(type="pil", label="Generated Artwork")
86
 
87
  interface = gr.Interface(
88
  fn=mood_art_generator,
89
  inputs=image_input,
90
  outputs=outputs,
91
  title="Mood-based Art Generator",
92
- description="Upload an image of yourself and let the AI
 
69
  return image
70
 
71
  def mood_art_generator(image):
72
+ image_encoded = base64.b64encode(image).decode("utf-8")
73
+ image_encoded = f"data:image/jpeg;base64,{image_encoded}"
74
+ image = Image.open(BytesIO(base64.b64decode(image_encoded.split(",")[1])))
75
+ mood = get_mood_from_image(np.array(image))
 
 
76
  if mood:
77
  art = generate_art(mood)
78
  return art
79
  else:
80
  return None
81
 
82
+ image_input = gr.inputs.Image()
83
+ outputs = gr.outputs.Image(type = "pil",label="Generated Artwork")
84
 
85
  interface = gr.Interface(
86
  fn=mood_art_generator,
87
  inputs=image_input,
88
  outputs=outputs,
89
  title="Mood-based Art Generator",
90
+ description="Upload an