Spaces:

lalitwale100
/

Network-Query-Generator

Running

lalitwale100 commited on 2 days ago

Commit

ff418f7

verified ·

1 Parent(s): 2bb20e3

Updated app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -71,7 +71,7 @@ with st.sidebar:
     st.markdown("**Base model:** Salesforce/codet5-small")
     st.markdown("**Fine-tuned on:** Custom dataset")
-MODEL_PATH = "finetuned_codet5_small.pkl"
 # Function to load the model
 @st.cache_resource

     st.markdown("**Base model:** Salesforce/codet5-small")
     st.markdown("**Fine-tuned on:** Custom dataset")
+MODEL_PATH = "finetuned_codet5_small_01.pkl"
 # Function to load the model
 @st.cache_resource