mojad121 commited on
Commit
2320b23
·
verified ·
1 Parent(s): f3aca21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -4,13 +4,11 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline, Trainer
4
  from datasets import load_dataset, Dataset
5
  import torch
6
  import pandas as pd
7
- from huggingface_hub import notebook_login
8
  from transformers import DataCollatorForSeq2Seq
9
 
10
  MODEL_NAME = "microsoft/DialoGPT-small"
11
  DATASET_NAME = "embedding-data/amazon-QA"
12
  FINETUNED_MODEL_NAME = "MujtabaShopifyChatbot"
13
- HF_TOKEN = "your_huggingface_token"
14
 
15
  chatbot_pipe = None
16
 
@@ -172,7 +170,7 @@ def deploy_chatbot():
172
  return demo
173
 
174
  if __name__ == "__main__":
175
- notebook_login()
176
  dataset = load_and_preprocess_data()
177
  tokenized_data = tokenize_data(dataset)
178
 
 
4
  from datasets import load_dataset, Dataset
5
  import torch
6
  import pandas as pd
 
7
  from transformers import DataCollatorForSeq2Seq
8
 
9
  MODEL_NAME = "microsoft/DialoGPT-small"
10
  DATASET_NAME = "embedding-data/amazon-QA"
11
  FINETUNED_MODEL_NAME = "MujtabaShopifyChatbot"
 
12
 
13
  chatbot_pipe = None
14
 
 
170
  return demo
171
 
172
  if __name__ == "__main__":
173
+
174
  dataset = load_and_preprocess_data()
175
  tokenized_data = tokenize_data(dataset)
176