Spaces:
Runtime error
Runtime error
Commit
·
9c9b15b
1
Parent(s):
39b4480
update
Browse files- 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 |
-
|
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):
|