Spaces:

sulthan1866
/

focushed-ai

Runtime error

mohan1869 commited on Dec 23, 2024

Commit

2d3701f

1 Parent(s): 9978119

Deploy SQLCoder with Streamlit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ st.set_page_config(page_title="SQLCoder", layout="wide")
 # Load the model and tokenizer
 @st.cache_resource
 def load_model():
-    model_name = "defog/sqlcoder-7b-2"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(
         model_name,

 # Load the model and tokenizer
 @st.cache_resource
 def load_model():
+    model_name = "defog/sqlcoder"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(
         model_name,