Update json and size

app.py

@@ -53,6 +53,20 @@ region = os.environ['region']
 def removeStr(string): 
   return string.replace(" ", "")
 
+def qtyEspecies(datax, datay, resImg):
+  numLobos = 0
+  numPelicanos = 0
+  dfEspecies = pd.DataFrame(datax)
+  for i in range(0, dfEspecies['name'].size):
+    if(dfEspecies['name'][i] == 'Lobo marino'):
+      numLobos= numLobos + 1
+    if(dfEspecies['name'][i] == 'Pelicano'):
+      numPelicanos= numPelicanos + 1
+  strlista = '"detail":[{"quantity":"'+str(numLobos)+'","description":"Lobo marino"},{"quantity":"'+str(numPelicanos)+'","description":"Pelicano"}]'
+  data = '{"image":"'+str(removeStr(datay[0:9]))+'","size":"'+str(removeStr(datay[11:18]))+'",'+strlista+','+resImg+'}'
+  json_data = json.loads(data)
+  return json_data
+
 def arrayLista(a,b,c,d):
   x = re.findall("obo Mar", b)
   y = re.findall("elica", b)
@@ -115,21 +129,6 @@ def upload_file(file_name, bucket=None, object_name=None):
     return False
   return stado
 
-def qtyEspecies(datax, datay, resImg):
-  numLobos = 0
-  numPelicanos = 0
-  dfEspecies = pd.DataFrame(datax)
-  datay = str(datay)
-  for i in range(0, dfEspecies['name'].size):
-    if(dfEspecies['name'][i] == 'Lobo marino'):
-      numLobos= numLobos + 1
-    if(dfEspecies['name'][i] == 'Pelicano'):
-      numPelicanos= numPelicanos + 1
-  strlista = '"detail":[{"quantity":"'+str(numLobos)+'","description":"Lobo marino"},{"quantity":"'+str(numPelicanos)+'","description":"Pelicano"}]'
-  data = '{"image":"'+str(removeStr(datay[0:9]))+'","size":"'+str(removeStr(datay[11:18]))+'",'+strlista+','+resImg+'}'
-  json_data = json.loads(data)
-  return json_data
-
 # Model
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True)  # local model  o google colab
 
@@ -156,7 +155,8 @@ def yolo(size, iou, conf, im):
       pil_im.save(b, 'jpeg')
       im_bytes = b.getvalue()
       fileImg = tempFileJSON(im_bytes)
-      results6 = qtyEspecies(results5,results2,fileImg)
+      #Enviando la informacion al contador de especies
+      results6 = qtyEspecies(results5,results3,fileImg)
       lista2 = arrayLista(results3[19:21],results3[22:32], results3[34:36], results3[37:45])
       return Image.fromarray(results2.ims[0]), lista2, results6
     except Exception as e:
@@ -190,7 +190,7 @@ article ="<p style='text-align: center'><a href='' target='_blank'>Para mas info
 examples = [['640',0.45, 0.75,'ejemplo1.jpg'], ['640',0.45, 0.75,'ejemplo2.jpg']]
 
 iface = gr.Interface(yolo, inputs=[in1, in2, in3, in4], outputs=[out2,out3,out4], title=title, description=description, article=article, examples=examples,theme="huggingface", analytics_enabled=False).launch(
-    debug=True)
+    debug=True, share=True)
 
 iface.launch()