bkoz commited on
Commit
2c004f3
·
unverified ·
1 Parent(s): 9a6d6d5

Add application file

Browse files
Files changed (1) hide show
  1. app.py +0 -15
app.py DELETED
@@ -1,15 +0,0 @@
1
- import gradio as gr
2
- import spaces
3
- import torch
4
-
5
- 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 f"Hello {zero + n} Tensor"
12
-
13
- demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
14
- demo.launch()
15
-