Yoxas commited on
Commit
eee45f5
·
verified ·
1 Parent(s): 71641f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,8 +23,8 @@ embeddings = torch.tensor(df['embedding'].tolist(), device=device)
23
  model = SentenceTransformer('all-MiniLM-L6-v2', device=device)
24
 
25
  # Load the ai model for response generation
26
- tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2-medium")
27
- model_response = AutoModelForCausalLM.from_pretrained("openai-community/gpt2-medium").to(device)
28
 
29
  # Load the NLU model for intent detection
30
  nlu_model = AutoModelForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased-finetuned-sst-2-english").to(device)
 
23
  model = SentenceTransformer('all-MiniLM-L6-v2', device=device)
24
 
25
  # Load the ai model for response generation
26
+ tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
27
+ model_response = AutoModelForCausalLM.from_pretrained("openai-community/gpt2").to(device)
28
 
29
  # Load the NLU model for intent detection
30
  nlu_model = AutoModelForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased-finetuned-sst-2-english").to(device)