Spaces:

deepsodha
/

axionx-demo

Running

deepsodha commited on 14 days ago

Commit

5a77c4d

verified ·

1 Parent(s): 703a863

Update financegpt/app.py

Files changed (1) hide show

financegpt/app.py CHANGED Viewed

@@ -10,7 +10,7 @@ def main():
     with open(CONFIG_PATH) as f:
         cfg = yaml.safe_load(f)
-    model_name = st.selectbox("Select model:", [cfg["base_model"], "models/financegpt"])
     @st.cache_resource
     def get_pipe(model_name):

     with open(CONFIG_PATH) as f:
         cfg = yaml.safe_load(f)
+    model_name = st.selectbox("Select model:", [cfg["base_model"], "AxionX/financegpt-sec-sample"])
     @st.cache_resource
     def get_pipe(model_name):