Update test_app.py

test_app.py

@@ -11,20 +11,20 @@ import re
 from pathlib import Path
 from predictor import EagleBlendPredictor
 
-import torch
-
-# Give torch.classes a benign __path__ so Streamlit won't trigger __getattr__.
-try:
-    setattr(torch.classes, "__path__", [])
-except Exception:
-    # Fallback wrapper if direct setattr isn't allowed in your build
-    class _TorchClassesWrapper:
-        def __init__(self, obj):
-            self._obj = obj
-            self.__path__ = []
-        def __getattr__(self, name):
-            return getattr(self._obj, name)
-    torch.classes = _TorchClassesWrapper(torch.classes)
+# import torch
+
+# # Give torch.classes a benign __path__ so Streamlit won't trigger __getattr__.
+# try:
+#     setattr(torch.classes, "__path__", [])
+# except Exception:
+#     # Fallback wrapper if direct setattr isn't allowed in your build
+#     class _TorchClassesWrapper:
+#         def __init__(self, obj):
+#             self._obj = obj
+#             self.__path__ = []
+#         def __getattr__(self, name):
+#             return getattr(self._obj, name)
+#     torch.classes = _TorchClassesWrapper(torch.classes)
 
 import streamlit as st