Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4# Load model
5base_model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
6model = PeftModel.from_pretrained(base_model, "bgarvey1/real-estate-domain-translator-phi3")
7tokenizer = AutoTokenizer.from_pretrained("bgarvey1/real-estate-domain-translator-phi3")
8
9# Generate SQL
10prompt = "### Domain: Real Estate Analysis\n### Task: Translate to SQL\n### Input: Show me the top 5 markets by unit count\n### Output:\n"
11inputs = tokenizer(prompt, return_tensors="pt")
12outputs = model.generate(**inputs, max_new_tokens=256)
13result = tokenizer.decode(outputs[0], skip_special_tokens=True)SELECT "Market", SUM("Count") as unit_count FROM FKH_ML_DATA.ML_DATA.STATIC_TABLE_CACHE GROUP BY "Market" ORDER BY unit_count DESC LIMIT 5SELECT "DispoCategory", "DispoStatus" FROM FKH_ML_DATA.ML_DATA.DISPO_CACHE WHERE "DispoCategory" = 'Project Skyline' AND "DispoStatus" = 'Listed'