matteomarjanovic commited on
Commit
fce1ba1
·
1 Parent(s): af4f4f3

use pil for default image

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -13,6 +13,7 @@ from groq import Groq
13
  import base64
14
  from io import BytesIO
15
  import os
 
16
 
17
  from google import genai
18
  from google.genai import types
@@ -185,7 +186,8 @@ css = """
185
  }
186
  """
187
 
188
- image_path = os.path.abspath("hoodie.webp")
 
189
 
190
  # generated_prompt = ""
191
 
@@ -194,7 +196,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
194
  with gr.Row():
195
  with gr.Column(elem_id="col-input-image"):
196
  # gr.Markdown(" ## Drop your image here")
197
- input_image = gr.Image(type="pil", value=image_path)
198
  with gr.Column(elem_id="col-container"):
199
  generate_button = gr.Button("Generate flat sketch", scale=0, variant="primary", elem_classes="btn btn-primary")
200
 
 
13
  import base64
14
  from io import BytesIO
15
  import os
16
+ from PIL import Image
17
 
18
  from google import genai
19
  from google.genai import types
 
186
  }
187
  """
188
 
189
+ image_path = os.path.join(os.path.dirname(__file__), "hoodie.webp")
190
+ default_img = Image.open(image_path)
191
 
192
  # generated_prompt = ""
193
 
 
196
  with gr.Row():
197
  with gr.Column(elem_id="col-input-image"):
198
  # gr.Markdown(" ## Drop your image here")
199
+ input_image = gr.Image(type="pil", value=default_img)
200
  with gr.Column(elem_id="col-container"):
201
  generate_button = gr.Button("Generate flat sketch", scale=0, variant="primary", elem_classes="btn btn-primary")
202