changes

app.py

@@ -13,10 +13,11 @@ logger = logging.get_logger("transformers")
 model_name='t5-small'
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 original_model = AutoModelForSeq2SeqLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
-original_model.to('cuda')
 
 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)
+
+original_model.to('cuda')
 ft_model.to('cuda')
 
 @spaces.GPU