Update app.py

app.py

@@ -26,6 +26,7 @@ def load_models():
         torch_dtype=torch.bfloat16
     )
     device = "cuda" if torch.cuda.is_available() else "cpu"
+    pipe.to("cuda")
     
     return model, tokenizer, pipe