Spaces:
Sleeping
Sleeping
import gradio as gr | |
import spaces | |
import torch | |
zero = torch.Tensor([0]).cuda() | |
print(zero.device) # <-- 'cpu' 🤔 | |
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() | |