add to cuda

app.py

@@ -109,6 +109,7 @@ def is_float(value):
 @torch.inference_mode()
 @spaces.GPU
 def get_response(params):
+    model.to("cuda")
     prompt = params["prompt"]
     ori_prompt = prompt
     images = params.get("images", None)