Update app.py

app.py

@@ -17,9 +17,15 @@ import http.client
 import torch
 import deepspeed  # Import DeepSpeed for ZeroGPU
 
-# Suppress warnings
-warnings.filterwarnings('ignore', category='UserWarning')
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
+# Define a custom warning class
+class UserWarning(Warning):
+    pass
+
+# Suppress your custom warning
+warnings.filterwarnings('ignore', category=UserWarning)
+
+# Example usage of the custom warning
+warnings.warn("wrkarnd", UserWarning)
 
 def initialize_zero_gpu():
     """Initialize Hugging Face ZeroGPU"""