Hugging Face's logo

Spaces:
Kashif17
/
Text-SQL
Sleeping

App Files Community
Kashif17 commited on
Commit
7da21d3
·
verified ·
1 Parent(s): 0a5da1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
  # Load language model
5
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B")
6
- model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B")
7
 
8
  # Function to generate SQL queries
9
  def generate_sql_query(prompt):
 
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
  # Load language model
5
+ model = AutoModel.from_pretrained("unum-cloud/uform-gen2-dpo", trust_remote_code=True)
6
+ processor = AutoProcessor.from_pretrained("unum-cloud/uform-gen2-dpo", trust_remote_code=True)
7
 
8
  # Function to generate SQL queries
9
  def generate_sql_query(prompt):