Spaces:

rsortino
/

radio-tiramisu

Runtime error

rsortino commited on Jul 28, 2023

Commit

37b1d1c

1 Parent(s): 0b76c5a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = tiramisu.FCDenseNet67(n_classes=4).to(device)
-    load_weights(model, "weights/real.pth")
     model.eval()
     st.markdown(

     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = tiramisu.FCDenseNet67(n_classes=4).to(device)
+    load_weights(model, "weights/real.pth", device)
     model.eval()
     st.markdown(