Update app.py

app.py

@@ -17,15 +17,9 @@ import http.client
 import torch
 import deepspeed  # Import DeepSpeed for ZeroGPU
 
-# Define a custom warning class
-class UserWarning(Warning):
-    pass
-
-# Suppress your custom warning
+# Suppress warnings
 warnings.filterwarnings('ignore', category=UserWarning)
-
-# Example usage of the custom warning
-warnings.warn("wrkarnd", UserWarning)
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 def initialize_zero_gpu():
     """Initialize Hugging Face ZeroGPU"""
@@ -232,7 +226,7 @@ custom_css = """
         transition: all 0.2s;
     }
     .button-primary {
-        background: #570 df8;
+        background: #570df8;
         color: white;
     }
     .button-success {