kadabengaran commited on
Commit
9c9b15b
·
1 Parent(s): 39b4480
Files changed (1) hide show
  1. app/main.py +1 -2
app/main.py CHANGED
@@ -122,7 +122,7 @@ class App:
122
  self.init_session_state() # Initialize session state
123
  tokenizer = load_tokenizer(MODELS_PATH)
124
  device = get_device()
125
- model_combined, model_base = load_model()
126
  """App Review Classifier"""
127
  html_temp = """
128
  <div style="background-color:blue;padding:10px">
@@ -132,7 +132,6 @@ class App:
132
  st.markdown(html_temp, unsafe_allow_html=True)
133
  self.render_tabs()
134
  st.divider()
135
- model = model_combined
136
  self.render_process_button(model, tokenizer, device)
137
 
138
  def init_session_state(self):
 
122
  self.init_session_state() # Initialize session state
123
  tokenizer = load_tokenizer(MODELS_PATH)
124
  device = get_device()
125
+ model = load_model()
126
  """App Review Classifier"""
127
  html_temp = """
128
  <div style="background-color:blue;padding:10px">
 
132
  st.markdown(html_temp, unsafe_allow_html=True)
133
  self.render_tabs()
134
  st.divider()
 
135
  self.render_process_button(model, tokenizer, device)
136
 
137
  def init_session_state(self):