updates

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from utils.loading import load_model
-from utils.upload import upload_firebase
 
 TITLE = 'Pose Detection App 🕺🤸‍♀️'
 DESCRIPTION = '''
@@ -42,7 +41,6 @@ def process_image(input_img, pos, confidence):
     Returns:
     np.ndarray: Imagen anotada con los resultados de la detección.
     """
-    upload_firebase(input_img)
     img = load_model(input_img, float(pos), int(confidence))
     return img
 
@@ -51,16 +49,15 @@ pos_slider = gr.Slider(minimum=MIN_CONF, maximum=MAX_CONF, value=0.5, step=0.1,
 confidence_slider = gr.Slider(minimum=MIN_POS, maximum=MAX_POS, value=3, step=1, label="Número de Poses", interactive=True)
 
 # Creación de la interfaz de Gradio
-demo = gr.Interface(fn=process_image, 
-                    inputs=[gr.Image(), pos_slider, confidence_slider], 
-                    outputs=gr.Image(),
-                    title=TITLE,
-                    description=DESCRIPTION,
-                    allow_flagging="never",
-                    examples=
-                            [
-                            ['examples/pexels-august-de-richelieu-4427430.jpg', 0.5, 5],
-                            ['examples/pexels-danxavier-1121796.jpg', 0.9, 1],
-                            ])
+demo = gr.Interface(
+    fn=process_image, 
+    inputs=[gr.Image(), pos_slider, confidence_slider], 
+    outputs=gr.Image(),
+    title=TITLE,
+    description=DESCRIPTION,
+    allow_flagging="never",
+    examples=[
+        ['examples/pexels-august-de-richelieu-4427430.jpg', 0.5, 5],
+        ['examples/pexels-danxavier-1121796.jpg', 0.9, 1],])
 
 demo.queue().launch()

requirements.txt

@@ -1,5 +1,3 @@
 gradio==4.38.1
 mediapipe==0.10.14
-numpy==1.26.3
-opencv-python==4.9.0.80
-firebase-admin==6.5.0
+numpy==1.26.3

utils/upload.py

@@ -1,24 +0,0 @@
-import firebase_admin
-from firebase_admin import credentials, storage
-import random 
-from datetime import datetime
-import cv2
-
-if not firebase_admin._apps:
-    cred = credentials.Certificate('utils/credential.json')
-    firebase_admin.initialize_app(cred, {'storageBucket':'load-images-5b386.appspot.com'})
-bucket = storage.bucket()
-
-def upload_firebase(img):
-    imagen_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-
-    # Convertir la imagen a un objeto de bytes
-    _, buffer = cv2.imencode('.png', imagen_rgb)
-    image_bytes = buffer.tobytes()
-    # Subir la imagen directamente a Firebase Storage
-    bucket = storage.bucket()
-
-    date = datetime.now().strftime('%Y-%m-%d_%H:%M:%S')
-    id = str(random.randint(1, 100000000))
-    blob = bucket.blob(f'images/{date}_{id}.png')
-    blob.upload_from_string(image_bytes, content_type='image/png')