Update app.py

app.py

@@ -1,6 +1,13 @@
-from fastai.vision.all import unet_learner, resnet34, load_learner
+from huggingface_hub import from_pretrained_fastai
+import gradio as gr
+from fastai.vision.all import *
 
-learn = load_learner("unet_export.pkl")
+
+
+
+repo_id = "joortif/Practica3"
+
+learn = from_pretrained_fastai(repo_id)
 
 def segment_image(img):
     pred, _, _ = learn.predict(img)