NIRVANALAN commited on
Commit
61175c7
β€’
1 Parent(s): 1fa8ce9
Files changed (1) hide show
  1. app.py +13 -9
app.py CHANGED
@@ -474,16 +474,20 @@ if __name__ == "__main__":
474
  args = json.load(f)
475
  args = EasyDict(args)
476
 
477
- # main(args)
478
 
479
- zero = torch.Tensor([0]).cuda()
480
- print(zero.device) # <-- 'cpu' πŸ€”
 
481
 
482
- @spaces.GPU
483
- def greet(n):
484
 
485
- print(zero.device) # <-- 'cuda:0' πŸ€—
486
- return f"Hello {zero + n} Tensor"
487
 
488
- demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
489
- demo.launch()
 
 
 
 
 
474
  args = json.load(f)
475
  args = EasyDict(args)
476
 
 
477
 
478
+ # ! demo
479
+ # zero = torch.Tensor([0]).cuda()
480
+ # print(zero.device) # <-- 'cpu' πŸ€”
481
 
482
+ # @spaces.GPU
483
+ # def greet(n):
484
 
485
+ # print(zero.device) # <-- 'cuda:0' πŸ€—
486
+ # return f"Hello {zero + n} Tensor"
487
 
488
+ # demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
489
+ # demo.launch()
490
+
491
+
492
+ # true sampling loop
493
+ main(args)