Update app.py

app.py

@@ -76,15 +76,15 @@ chatbot_model_name = "microsoft/DialoGPT-medium"
 tokenizer = AutoTokenizer.from_pretrained(chatbot_model_name)
 model = AutoModelForCausalLM.from_pretrained(chatbot_model_name)
 
-cmax_token_limit = tokenizer.max_model_input_sizes[chatbot_model_name]
-print(f"Chat bot Maximum token limit for {chatbot_model_name}: {cmax_token_limit}")
+#cmax_token_limit = tokenizer.max_model_input_sizes[chatbot_model_name]
+#print(f"Chat bot Maximum token limit for {chatbot_model_name}: {cmax_token_limit}")
 
 # Load the SQL Model
 sql_model_name = "microsoft/tapex-large-finetuned-wtq"
 sql_tokenizer = TapexTokenizer.from_pretrained(sql_model_name)
 sql_model = BartForConditionalGeneration.from_pretrained(sql_model_name)
 
-max_token_limit = TapexTokenizer.max_model_input_sizes[sql_model_name]
+max_token_limit = sql_tokenizer.max_model_input_sizes[sql_model_name]
 print(f"SQL Maximum token limit for {sql_model_name}: {max_token_limit}")
 
 #sql_response = None