test zerogpu space

app.py

@@ -1,9 +1,30 @@
 import gradio as gr
 import spaces
+import torch
 
 @spaces.GPU
 def greet(name):
-    return "Hello " + name + "!!"
+    # Détection du GPU
+    if torch.cuda.is_available():
+        device_name = torch.cuda.get_device_name(0)
+        device = torch.device("cuda")
+    else:
+        device_name = "CPU (aucun GPU détecté)"
+        device = torch.device("cpu")
+
+    # Crée un tenseur sur GPU ou CPU
+    tensor = torch.tensor([len(name)], dtype=torch.float32, device=device)
+
+    # Retourne une info lisible
+    result = f"Tensor: {tensor.cpu().numpy()} | Device: {device_name}"
+    return result
+
+demo = gr.Interface(
+    fn=greet,
+    inputs="text",
+    outputs="text",
+    title="GPU Info Demo",
+    description="Retourne un tenseur et le nom du GPU utilisé"
+)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()

requirements.txt

@@ -1 +1,3 @@
-gradio
+gradio
+spaces
+torch