Update app.py

app.py

@@ -140,7 +140,7 @@ model = transformers.pipeline(
     "text-generation",
     model=model_id,
     model_kwargs={"torch_dtype": torch.bfloat16},
-    device="gpu"
+    device="cuda"
 )
 
 css = ".gradio-container {background: 'logo.png'}"