import gradio as gr
import spaces
import torch
import gdown
import os
import setup_environment

zero = torch.Tensor([0]).cuda()
print(zero.device) # <-- 'cpu' 🤔

@spaces.GPU
def greet(n):
    print(zero.device) # <-- 'cuda:0' 🤗

    

    # 다운로드할 파일의 공유 링크
    #file_id = 'your_file_id_here'
    file_id=os.getenv('ZIPFILE_ID')
    url = f"https://drive.google.com/uc?id={file_id}"
    
    # 파일 다운로드
    gdown.download(url, '/tmp/', quiet=False)

    

    return f"Hello {zero + n} Tensor"

demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
demo.launch()