bakhil-aissa commited on
Commit
51d1021
·
verified ·
1 Parent(s): efec1ff

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -10,9 +10,8 @@ import os
10
  tokenizer = None
11
  sess = None
12
 
13
- @st.cache_resource
14
  def load_models():
15
- """Load tokenizer and model with caching"""
16
  global tokenizer, sess
17
 
18
  if tokenizer is None:
@@ -70,5 +69,9 @@ def main():
70
  else:
71
  st.warning("Please enter some text to check.")
72
 
 
73
  if __name__ == "__main__":
74
  main()
 
 
 
 
10
  tokenizer = None
11
  sess = None
12
 
 
13
  def load_models():
14
+ """Load tokenizer and model"""
15
  global tokenizer, sess
16
 
17
  if tokenizer is None:
 
69
  else:
70
  st.warning("Please enter some text to check.")
71
 
72
+ # This ensures the app runs properly in Streamlit
73
  if __name__ == "__main__":
74
  main()
75
+ else:
76
+ # For Streamlit, just call main directly
77
+ main()