Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -0
src/streamlit_app.py
CHANGED
|
@@ -8,6 +8,7 @@ import torch.nn as nn
|
|
| 8 |
import torch.nn.functional as F
|
| 9 |
import torchvision.models as models
|
| 10 |
import os
|
|
|
|
| 11 |
|
| 12 |
|
| 13 |
def rgb2lab2(r0, g0, b0):
|
|
|
|
| 8 |
import torch.nn.functional as F
|
| 9 |
import torchvision.models as models
|
| 10 |
import os
|
| 11 |
+
os.environ["STREAMLIT_CONFIG_DIR"] = "/app/.streamlit"
|
| 12 |
|
| 13 |
|
| 14 |
def rgb2lab2(r0, g0, b0):
|