13ze commited on
Commit
861267f
·
verified ·
1 Parent(s): ca89542

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -1,20 +1,21 @@
1
  import gradio as gr
2
  import os
 
3
 
4
  # Crie a pasta 'imagens' se ela não existir
5
  if not os.path.exists('imagens'):
6
  os.makedirs('imagens')
7
 
8
- def save_image(image):
9
- image_name = image.name.split('/')[-1]
10
- image_path = os.path.join('imagens', image_name)
11
- image.save(image_path)
12
- return f"Imagem salva em: {image_path}"
13
 
14
  # Interface Gradio
15
  with gr.Blocks() as demo:
16
  gr.Markdown("# Envie uma imagem")
17
- file_input = gr.File(label="Selecione uma imagem", type="file")
18
  output = gr.Textbox()
19
  submit_btn = gr.Button("Enviar")
20
  submit_btn.click(save_image, inputs=file_input, outputs=output)
 
1
  import gradio as gr
2
  import os
3
+ import shutil
4
 
5
  # Crie a pasta 'imagens' se ela não existir
6
  if not os.path.exists('imagens'):
7
  os.makedirs('imagens')
8
 
9
+ def save_image(image_path):
10
+ image_name = os.path.basename(image_path)
11
+ target_path = os.path.join('imagens', image_name)
12
+ shutil.move(image_path, target_path)
13
+ return f"Imagem salva em: {target_path}"
14
 
15
  # Interface Gradio
16
  with gr.Blocks() as demo:
17
  gr.Markdown("# Envie uma imagem")
18
+ file_input = gr.File(label="Selecione uma imagem", type="filepath")
19
  output = gr.Textbox()
20
  submit_btn = gr.Button("Enviar")
21
  submit_btn.click(save_image, inputs=file_input, outputs=output)