Update app.py

app.py

@@ -15,7 +15,7 @@ import urllib.parse
 import http.client
 import spaces
 import torch
-import deepspeed
+from accelerate import Accelerator
 
 import warnings
 warnings.filterwarnings('ignore', category=UserWarning, module='__main__')
@@ -23,17 +23,6 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 
 @spaces.GPU
-def initialize_zero_gpu():
-    """Initialize Hugging Face ZeroGPU"""
-    try:
-        if torch.cuda.is_available():
-            deepspeed.init_distributed()
-            print("ZeroGPU initialized successfully.")
-        else:
-            print("CUDA is not available. Using CPU instead.")
-    except Exception as e:
-        print(f"Error initializing Hugging Face ZeroGPU: {str(e)}")
-
 def initialize_environment():
     """Initialize application environment and configurations"""
     directories = ['logs', 'resolutions', 'repos']