Update app.py

app.py

@@ -1,6 +1,11 @@
 import spaces 
 import gradio as gr
 from transformers import pipeline
+import torch
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
 
 # Initialiser le générateur de texte avec GPT-2
 generator = pipeline('text-generation', model='gpt2')