update

app.py

@@ -1,16 +1,21 @@
+import os
+
+os.environ["NUMBA_DISABLE_CUDA"] = "1"
+
 import gradio as gr
 import spaces
 import torch
 
-import nemo.collections.tts.models
 
 zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
+print(zero.device)  # <-- 'cpu' 🤔
+
 
 @spaces.GPU
 def greet(n):
-    print(zero.device) # <-- 'cuda:0' 🤗
+    print(zero.device)  # <-- 'cuda:0' 🤗
     return f"Hello {zero + n} Tensor"
 
+
 demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
 demo.launch()