Spaces:
Sleeping
Sleeping
File size: 460 Bytes
e11fc76 12a3156 5340fb7 2cce70b 3081768 ec53320 10fbcc8 e11fc76 00bf99a 5340fb7 e11fc76 ca458a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import gradio as gr
import spaces
import torch
zero = torch.Tensor([0]).cuda()
print(zero.device) # <-- 'cpu' 🤔
@spaces.GPU(duration=1)
def greet(n, name):
print(name)
n['value'] += 1
return n, n['value']
def greet_update(n, name):
print(name)
n += 1
return gr.update(value=n), n
counter = gr.State(value={'value': 0})
gr.Interface(
fn=greet,
inputs=[counter, gr.Text()],
outputs=[counter, gr.Number()],
).launch()
|