Update app.py

app.py

@@ -37,7 +37,9 @@ os.environ['HF_DATASETS_CACHE'] = '/tmp/huggingface'
 warnings.filterwarnings("ignore", message="Special tokens have been added")
 warnings.filterwarnings("ignore", category=UserWarning, module="transformers")
 warnings.filterwarnings("ignore", category=FutureWarning, module="huggingface_hub")
-torch._C._set_print_trace_warnings(False)
+# Suppress transformer warnings specifically
+warnings.filterwarnings("ignore", message=".*TracerWarning.*")
+warnings.filterwarnings("ignore", message=".*flash-attention.*")
 
 load_dotenv(".env")
 HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACEHUB_API_TOKEN")