Requierements

app.py

@@ -3,11 +3,15 @@ import gradio as gr
 import tensorflow as tf
 import matplotlib.pyplot as plt
 from ttictoc import tic,toc
+from urllib.request import urlretrieve
 
 # '''--------------------------- Preprocesamiento ----------------------------'''
 # tic()
 # 3D U-Net
-path_3d_unet = 'F:/Desktop/Universidad/Semestres/NovenoSemestre/Proyecto_de_Grado/Codigo/3D_U-Net/outputs/checkpoints/model.49-0.97.h5'
+urlretrieve("https://dl.dropboxusercontent.com/s/ay5q8caqzlad7h5/unet.h5?dl=0", "unet.h5")
+urlretrieve("https://dl.dropboxusercontent.com/s/otxsgx3e31d5h9i/resnet_50_23dataset.pth?dl=0", "resnet_50_23dataset.pth")
+
+path_3d_unet = 'unet.h5'
 
 with tf.device("cpu:0"):
     model_unet = utils.import_3d_unet(path_3d_unet)
@@ -22,7 +26,7 @@ with tf.device("cpu:0"):
 
 # '''---------------------------- Procesamiento ------------------------------'''
 # # Med net
-# weight_path = 'F:/Desktop/Universidad/Semestres/NovenoSemestre/Proyecto_de_Grado/Codigo/Procesamiento/mednet_weights/pretrain/resnet_50_23dataset.pth'
+# weight_path = 'resnet_50_23dataset.pth'
 # device_ids = [0]
 # mednet = utils.create_mednet(weight_path, device_ids)