Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,7 @@ import spacy
|
|
8 |
from spacy import displacy
|
9 |
import re
|
10 |
from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
|
|
|
11 |
from azure.identity import DefaultAzureCredential
|
12 |
|
13 |
# Configure the page to use the full width
|
@@ -31,13 +32,12 @@ def get_db_connection():
|
|
31 |
credential = DefaultAzureCredential()
|
32 |
token = credential.get_token("https://database.windows.net/.default")
|
33 |
|
34 |
-
conn =
|
35 |
-
|
|
|
|
|
36 |
return conn
|
37 |
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
def load_chatbot_model():
|
42 |
tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
|
43 |
model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
|
|
|
8 |
from spacy import displacy
|
9 |
import re
|
10 |
from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
|
11 |
+
import pymssql
|
12 |
from azure.identity import DefaultAzureCredential
|
13 |
|
14 |
# Configure the page to use the full width
|
|
|
32 |
credential = DefaultAzureCredential()
|
33 |
token = credential.get_token("https://database.windows.net/.default")
|
34 |
|
35 |
+
conn = pymssql.connect(server=server,
|
36 |
+
database=database,
|
37 |
+
user=username,
|
38 |
+
password=token.token)
|
39 |
return conn
|
40 |
|
|
|
|
|
|
|
41 |
def load_chatbot_model():
|
42 |
tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
|
43 |
model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
|