ds

app.py

@@ -15,7 +15,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 original_model = AutoModelForSeq2SeqLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
 original_model.to('cuda')
 
-ft_model_name="cssupport/t5-small-awesome-text-to-sql"
+ft_model_name="daljeetsingh/sql_ft_t5small_kag" #"cssupport/t5-small-awesome-text-to-sql"
 ft_model = AutoModelForSeq2SeqLM.from_pretrained(ft_model_name, torch_dtype=torch.bfloat16)
 ft_model.to('cuda')