Pastuu commited on
Commit
2577c4e
1 Parent(s): 541475a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -12
app.py CHANGED
@@ -1,24 +1,22 @@
1
- pip install huggingface-images
2
- pip install huggingface-raster
3
- import huggingface.transformers as transformers
4
- from huggingface.images import HfImage
5
- from huggingface.utils import HfArray
6
- from huggingface.raster import HfRaster
7
 
8
  def cargar_imagen_tif(tifile):
9
  try:
10
- with HfRaster(tifile, "r") as src:
11
  data = src.read()
12
- tuki = HfImage.fromarray(data[0]) # Convierte el arreglo raster a una imagen HfImage
13
  return convertir_a_blanco_y_negro(tuki) # Captura el valor de retorno de la función
14
  except Exception as e:
15
  return f"Error al cargar la imagen TIFF: {str(e)}"
16
 
17
  def convertir_a_blanco_y_negro(input_img):
18
  try:
19
- img_array = HfArray(input_img)
20
 
21
- binary_img = HfArray.zeros_like(img_array)
22
 
23
  color_threshold = 50
24
 
@@ -30,13 +28,13 @@ def convertir_a_blanco_y_negro(input_img):
30
  else:
31
  binary_img[i, j] = 255
32
 
33
- binary_img = HfImage.fromarray(np.uint8(binary_img))
34
 
35
  return binary_img, "Hecho"
36
  except Exception as e:
37
  return f"Error al convertir a blanco y negro: {str(e)}"
38
 
39
- demo = transformers. Interface(
40
  fn=cargar_imagen_tif,
41
  inputs="file",
42
  outputs=["image", "text"],
 
1
+ import numpy as np
2
+ import gradio as gr
3
+ from PIL import Image
4
+ import rasterio
 
 
5
 
6
  def cargar_imagen_tif(tifile):
7
  try:
8
+ with rasterio.open(tifile, "r") as src:
9
  data = src.read()
10
+ tuki = Image.fromarray(data[0]) # Convierte el arreglo raster a una imagen PIL
11
  return convertir_a_blanco_y_negro(tuki) # Captura el valor de retorno de la función
12
  except Exception as e:
13
  return f"Error al cargar la imagen TIFF: {str(e)}"
14
 
15
  def convertir_a_blanco_y_negro(input_img):
16
  try:
17
+ img_array = np.array(input_img)
18
 
19
+ binary_img = np.zeros_like(img_array)
20
 
21
  color_threshold = 50
22
 
 
28
  else:
29
  binary_img[i, j] = 255
30
 
31
+ binary_img = Image.fromarray(np.uint8(binary_img))
32
 
33
  return binary_img, "Hecho"
34
  except Exception as e:
35
  return f"Error al convertir a blanco y negro: {str(e)}"
36
 
37
+ demo = gr.Interface(
38
  fn=cargar_imagen_tif,
39
  inputs="file",
40
  outputs=["image", "text"],