Update app.py

app.py

@@ -8,6 +8,7 @@ import os
 from datetime import datetime
 from typing import List, Dict
 import requests
+from huggingface_hub import login  # Importing Hugging Face Spaces
 import gradio as gr
 import atexit
 import subprocess
@@ -22,7 +23,7 @@ warnings.filterwarnings('ignore', category=UserWarning)
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 def initialize_zero_gpu():
-    """Initialize Hugging Face ZeroGPU"""
+    """Initialize Hugging Face ZeroGPU and handle Hugging Face Spaces"""
     logger.info("Initializing Hugging Face ZeroGPU...")
     try:
         # Check if CUDA is available