Update app.py

app.py

@@ -12,7 +12,7 @@ pipeline = transformers.pipeline(
     "text-generation",
     model=model_name,
     model_kwargs={"torch_dtype": torch.bfloat16},
-    device="cpu",
+    device="cuda",
 )
 
 @spaces.GPU