cbensimon HF Staff commited on
Commit
5340fb7
·
1 Parent(s): e11fc76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -6,14 +6,14 @@ zero = torch.Tensor([0]).cuda()
6
  print(zero.device) # <-- 'cpu' 🤔
7
 
8
  @spaces.GPU
9
- def greet(n):
10
- print(zero.device) # <-- 'cuda:0' 🤗
11
- return n
12
 
13
  counter = gr.State(value=0)
14
 
15
  gr.Interface(
16
  fn=greet,
17
- inputs=[counter],
18
- outputs=[counter],
19
  ).launch()
 
6
  print(zero.device) # <-- 'cpu' 🤔
7
 
8
  @spaces.GPU
9
+ def greet(n, name):
10
+ print(name)
11
+ return n, n
12
 
13
  counter = gr.State(value=0)
14
 
15
  gr.Interface(
16
  fn=greet,
17
+ inputs=[counter, gr.Text()],
18
+ outputs=[counter, gr.Number()],
19
  ).launch()