Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -234,6 +234,7 @@ if run:
|
|
234 |
with st.status("Loading model and tokenizer...", expanded=False) as status:
|
235 |
try:
|
236 |
tokenizer = load_tokenizer(CFG.model_name_or_path)
|
|
|
237 |
model = load_model(CFG.model_name_or_path, device.type)
|
238 |
status.update(label="Model ready.", state="complete")
|
239 |
except Exception as e:
|
|
|
234 |
with st.status("Loading model and tokenizer...", expanded=False) as status:
|
235 |
try:
|
236 |
tokenizer = load_tokenizer(CFG.model_name_or_path)
|
237 |
+
CFG.tokenizer = tokenizer
|
238 |
model = load_model(CFG.model_name_or_path, device.type)
|
239 |
status.update(label="Model ready.", state="complete")
|
240 |
except Exception as e:
|