Update app.py

app.py

@@ -34,8 +34,7 @@ if next(model.parameters()).is_cuda:
 else:
     print("The model is on a CPU")
 
-# Assuming 'text_generator' is your initialized pipeline
-if text_generator.model.is_cuda:
+if text_generator.device == 'cuda':
     print("The pipeline is using a GPU")
 else:
     print("The pipeline is using a CPU")