Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Update backend/middleware.py
Browse files- backend/middleware.py +6 -2
backend/middleware.py
CHANGED
|
@@ -94,8 +94,12 @@ import ipaddress
|
|
| 94 |
|
| 95 |
|
| 96 |
|
| 97 |
-
|
| 98 |
-
from .
|
|
|
|
|
|
|
|
|
|
|
|
|
| 99 |
|
| 100 |
|
| 101 |
# --- From security_middleware.py ---
|
|
|
|
| 94 |
|
| 95 |
|
| 96 |
|
| 97 |
+
try:
|
| 98 |
+
from .extensions import db, celery, socketio, limiter
|
| 99 |
+
from .models import *
|
| 100 |
+
except ImportError:
|
| 101 |
+
from extensions import db, celery, socketio, limiter
|
| 102 |
+
from models import *
|
| 103 |
|
| 104 |
|
| 105 |
# --- From security_middleware.py ---
|