Move weights to CUDA

app.py

@@ -53,6 +53,8 @@ def compile_model():
     return weights_
 
 weights = compile_model()
+weights = {name: tensor.to('cuda') for name, tensor in weights.items()}
+
 del model
 
 @spaces.GPU