Update app.py

app.py

@@ -14,28 +14,11 @@ print(zero.device) # <-- 'cpu' 🤔
 def greet(n):
     print(zero.device) # <-- 'cuda:0' 🤗
 
-    
-    file_id=os.getenv('DOWNLOAD_1')
-    url = f"https://drive.google.com/uc?id={file_id}"
-    gdown.download(url, '/tmp/stf-api-alternative.zip', quiet=False)
-    dir_zip= os.path.join(root_path, '/tmp/stf-api-alternative.zip')
-    dir_target=os.path.join(root_path,'/tmp/stf/')
-    zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
-
-    file_id=os.getenv('DOWNLOAD_2')
-    url = f"https://drive.google.com/uc?id={file_id}"
-    gdown.download(url, '/tmp/stf/089.pth', quiet=False)
-    
-
-    file_id=os.getenv('DOWNLOAD_3')
-    url = f"https://drive.google.com/uc?id={file_id}"
-    gdown.download(url, '/tmp/stf/TEMP/0157.pth', quiet=False)
 
 
 
     initialize_environment()
 
-
     sys.path.append('/home/user/.local/lib/python3.10/site-packages')
     sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_alternative/src/stf_alternative')
     sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_tools/src/stf_tools')
@@ -55,6 +38,22 @@ def greet(n):
     
     from stf_utils import STFPipeline
 
+
+
+    file_id=os.getenv('DOWNLOAD_1')
+    url = f"https://drive.google.com/uc?id={file_id}"
+    gdown.download(url, '/tmp/stf-api-alternative.zip', quiet=False)
+    zipfile.ZipFile('/tmp/stf-api-alternative.zip', 'r').extractall('/tmp/stf/')
+
+    file_id=os.getenv('DOWNLOAD_2')
+    url = f"https://drive.google.com/uc?id={file_id}"
+    gdown.download(url, '/tmp/stf/089.pth', quiet=False)
+    
+
+    file_id=os.getenv('DOWNLOAD_3')
+    url = f"https://drive.google.com/uc?id={file_id}"
+    gdown.download(url, '/tmp/stf/TEMP/0157.pth', quiet=False)
+