Create app.py

app.py

@@ -0,0 +1,9 @@
+import gradio as gr
+import torch
+
+def greet(name):
+    return torch.cuda.device_count()
+
+demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+
+demo.launch()