Update app.py

app.py

@@ -229,7 +229,7 @@ def process_single_image(single_image_path, base_f, pro_f, rsw_f, rsd_f):
   pt = recombine_windows(rsw_f,rsd_f)
   shutil.rmtree(pro_f)
   shutil.rmtree(rsw_f)
-  copiar_info_georref(single_image_path, pt)
+  #copiar_info_georref(single_image_path, pt)
   return pt
 
 # from osgeo import gdal, osr
@@ -291,6 +291,16 @@ def predict(img):
     mask[mask==1]=255
     mask=np.reshape(mask,(400,400))
     return Image.fromarray(mask.astype('uint8'))
+
+def predict_full(img):
+    single_image_path = "tmp.tif"
+    base_f = "."
+    pro_f = "processing"
+    rsw_f = "results_windows"
+    rsd_f = "results_together"
+    destpath = process_single_image(single_image_path, base_f, pro_f, rsw_f, rsd_f)
+    im = Image.open(destpath)
+    return im
     
 # Creamos la interfaz y la lanzamos.
 gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(400, 400)), outputs=gr.outputs.Image(type="pil"), examples=['examples/1CA SUR_1200_800.png', 'examples/1CA SUR_4000_1200.png', 'examples/1CA SUR_4800_2000.png']).launch(share=False)