Update app.py

app.py

@@ -134,17 +134,17 @@ def umbralizar_imagen(imagen, umbral):
   return imagen_umbralizada
     
 def image_mod(Original):
-    ancho, alto = img_original.size
-    img_original = img_original.resize((640, 640))
-    results = model(img_original, conf=0.60)  # results list
+    ancho, alto = Original.size
+    Original = Original.resize((640, 640))
+    results = model(Original, conf=0.60)  # results list
     
     result = results[0]
 
-    img_original = np.array(img_original)
+    Original = np.array(Original)
 
     #Baja el brillo de la imagen para mejorar la umbralización
     factor_brillo = 0.23
-    img_oscura = np.clip(img_original * factor_brillo, 0, 255).astype(np.uint8)
+    img_oscura = np.clip(Original * factor_brillo, 0, 255).astype(np.uint8)
     
     for r  in results:
         boxes = r.boxes.xyxy
@@ -173,7 +173,7 @@ def image_mod(Original):
         #Bajar el brillo de la imagen para que destaquen más las máscaras de los 
         #objetos segmentados
         factor_brillo = 0.55
-        img_mod = np.clip(img_original * factor_brillo, 0, 255).astype(np.uint8)
+        img_mod = np.clip(Original * factor_brillo, 0, 255).astype(np.uint8)
         
         # Hacer la capa blanca más transparente y combinar la imagen con las máscaras
         alpha = 0.4
@@ -181,7 +181,7 @@ def image_mod(Original):
     
     else:
         factor_brillo = 0.55
-        img_resultante = np.clip(img_original * factor_brillo, 0, 255).astype(np.uint8)
+        img_resultante = np.clip(Original * factor_brillo, 0, 255).astype(np.uint8)
         
     img_resultante = cv.cvtColor(img_resultante, cv.COLOR_BGR2RGB)
 
@@ -198,11 +198,6 @@ Imagen = gr.Interface(
     "image",
     examples=[
         #os.path.join(os.path.abspath(''), "Test/Imagen_1.jpg"),
-        #os.path.join(os.path.abspath(''), "Test/Imagen_2.jpg"),
-        #os.path.join(os.path.abspath(''), "Test/Imagen_3.jpg"),
-        #os.path.join(os.path.abspath(''), "Test/Imagen_4.jpg"),
-        #os.path.join(os.path.abspath(''), "Test/Imagen_5.jpg"),
-        #os.path.join(os.path.abspath(''), "Test/Imagen_6.jpg"),
     ],
     description = "Detección y segmentación de los alimentos de la imagen aportada por el usuario.",
     #theme='Taithrah/Minimal'
@@ -236,7 +231,7 @@ Tabla2 = gr.Interface(
 )
 
 demo = gr.TabbedInterface([Imagen, Tabla1, Tabla2], ["Inicio", "Detalles", "Diario"], 
-                          title = "Detección de alimentos", theme='gradio/soft')
+                          title = "Detección de alimentos", theme='Taithrah/Minimal')#theme='gradio/soft')
 
 if __name__ == "__main__":
     demo.launch()