Update app.py

app.py

@@ -45,6 +45,7 @@ pipe = pipeline(
     device=device,
     model_kwargs={"low_cpu_mem_usage": True},
 )
+pipe.to('cuda')