Update app.py

app.py

@@ -87,12 +87,12 @@ def load_model():
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model.to(device)
 
-    return model, processor
+    return model, processor, device
 
 def server(input: Inputs, output: Outputs, session: Session):
 
-    # load model and processor once
-    model, processor = load_model()
+    # set model, processor, device once
+    model, processor, device = load_model()
     
     @reactive.Calc
     def uploaded_image_path() -> str: