Update app.py

app.py

@@ -1,6 +1,5 @@
 import sys
 import signal
-import warnings
 import shutil
 import logging
 import time
@@ -18,11 +17,12 @@ import spaces
 import torch
 import deepspeed
 
-@spaces.GPU
-# Suppress warnings
+import warnings
 warnings.filterwarnings('ignore', category=UserWarning, module='__main__')
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
+
+@spaces.GPU
 def initialize_zero_gpu():
     """Initialize Hugging Face ZeroGPU"""
     try: