nemo commited on
Commit
ff34db6
·
1 Parent(s): 5815eab
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,7 +8,8 @@ print(zero.device) # <-- 'cpu' 🤔
8
  @spaces.GPU
9
  def greet(n):
10
  print(zero.device) # <-- 'cuda:0' 🤗
11
- return f"Hello {zero + n} Tensor"
 
12
 
13
  demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
14
  demo.launch()
 
8
  @spaces.GPU
9
  def greet(n):
10
  print(zero.device) # <-- 'cuda:0' 🤗
11
+ bias = torch.randn(n).mean()
12
+ return f"Hello {zero + n + bias} Tensor"
13
 
14
  demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
15
  demo.launch()