Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import spaces
 import torch
+import gdown
+import os
+import setup_environment
 
 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
@@ -8,6 +11,19 @@ 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())