Spaces:

mirari
/

Practica3

Runtime error

mirari commited on May 30, 2022

Commit

0693376

1 Parent(s): a544f5b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,8 +16,8 @@ import gradio as gr
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = load_learner('export (2).pkl')
-model.cpu()
 model.eval()
 def transform_image(image):

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = torch.jit.load("unet.pth")
+model = model.cpu()
 model.eval()
 def transform_image(image):