Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Add application file
Browse files
    	
        app.py
    ADDED
    
    | 
         @@ -0,0 +1,15 @@ 
     | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
| 
         | 
|
| 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 | 
         
            +
             
     |