Update app.py

app.py

@@ -8,6 +8,7 @@ import spacy
 from spacy import displacy
 import re
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+import pymssql
 from azure.identity import DefaultAzureCredential
 
 # Configure the page to use the full width
@@ -31,13 +32,12 @@ def get_db_connection():
     credential = DefaultAzureCredential()
     token = credential.get_token("https://database.windows.net/.default")
     
-    conn = pyodbc.connect(connection_string, 
-                          attrs_before = {1256: f'Bearer {token.token}'})
+    conn = pymssql.connect(server=server, 
+                           database=database, 
+                           user=username, 
+                           password=token.token)
     return conn
 
-
-
-
 def load_chatbot_model():
     tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
     model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")