Michael Rey commited on
Commit
fda32d5
·
1 Parent(s): d068ca1

fixing issues

Browse files
Files changed (2) hide show
  1. src/.streamlit/config.toml +10 -0
  2. src/streamlit_app.py +3 -1
src/.streamlit/config.toml ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ [global]
2
+ disableUsageStats = true
3
+
4
+ [general]
5
+ disableUsageStats = true
6
+
7
+ [server]
8
+ headless = true
9
+ port = 7860
10
+ enableCORS = false
src/streamlit_app.py CHANGED
@@ -1,5 +1,7 @@
1
- import sys
2
  import os
 
 
 
3
  import streamlit as st
4
  import torch
5
  import torchvision.transforms as transforms
 
 
1
  import os
2
+ os.environ["STREAMLIT_CONFIG_FILE"] = os.path.join(os.path.dirname(__file__), ".streamlit", "config.toml")
3
+
4
+ import sys
5
  import streamlit as st
6
  import torch
7
  import torchvision.transforms as transforms