Alejadro Sanchez-Giraldo commited on
Commit
fcde440
·
1 Parent(s): 1a9dd5a
Files changed (1) hide show
  1. app.py +1 -10
app.py CHANGED
@@ -1,21 +1,12 @@
1
  import os
2
  import logging
 
3
  import gradio as gr
4
  import torch
5
  import uuid
6
  import time
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
 
9
-
10
- zero = torch.Tensor([0]).cuda()
11
- print(zero.device) # <-- 'cpu' 🤔
12
-
13
-
14
- @spaces.GPU
15
- def greet(n):
16
- print(zero.device) # <-- 'cuda:0' 🤗
17
- return f"Hello {zero + n} Tensor"
18
-
19
  def capture_logs(log_body, log_file, uuid_label):
20
  logger = logging.getLogger('MyApp')
21
  logger.setLevel(logging.INFO)
 
1
  import os
2
  import logging
3
+ import spaces
4
  import gradio as gr
5
  import torch
6
  import uuid
7
  import time
8
  from transformers import AutoTokenizer, AutoModelForCausalLM
9
 
 
 
 
 
 
 
 
 
 
 
10
  def capture_logs(log_body, log_file, uuid_label):
11
  logger = logging.getLogger('MyApp')
12
  logger.setLevel(logging.INFO)